Update Merge_K_SortedLinkedlist.java

format code
This commit is contained in:
yanglbme
2019-03-20 16:24:11 +08:00
committed by GitHub
parent f3ef5b64ae
commit 4bc83966c3

View File

@ -1,29 +1,25 @@
import java.util.*;
import java.util.Arrays;
import java.util.Comparator;
import java.util.PriorityQueue;
/**
* @author Arun Pandey (https://github.com/pandeyarun709)
*/
public class Merge_K_SortedLinkedlist {
/*
/**
* This function merge K sorted LinkedList
* @param Node []a : array of LinkedList
* @param int N : size of array
*
* author Arun Pandey (https://github.com/pandeyarun709)
* @param a array of LinkedList
* @param N size of array
* @return node
*/
Node mergeKList(Node[]a, int N)
{
Node mergeKList(Node[] a, int N) {
// Min Heap
PriorityQueue<Node> min = new PriorityQueue<>(new Comparator<Node>(){
public int compare(Node x , Node y){
return x.data - y.data;
}
});
PriorityQueue<Node> min = new PriorityQueue<>(Comparator.comparingInt(x -> x.data));
// adding head of all linkedList in min heap
for(int i =0 ; i< N ;i++)
{
min.add(a[i]);
}
min.addAll(Arrays.asList(a).subList(0, N));
// Make new head among smallest heads in K linkedList
Node head = min.poll();
@ -44,20 +40,15 @@ public class Merge_K_SortedLinkedlist {
}
return head;
}
// Node Class
private class Node {
private int data;
private Node next;
public Node(int d)
{
public Node(int d) {
this.data = d;
next = null;
}
}
}