algorithm/Java
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Java.git synced 2025-07-07 09:45:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update Merge_K_SortedLinkedlist.java

Browse Source 
format code
This commit is contained in:
yanglbme
2019-03-20 16:24:11 +08:00
committed by GitHub
parent f3ef5b64ae
commit 4bc83966c3
1 changed files with 37 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
DataStructures/Lists/Merge_K_SortedLinkedlist.java
View File

@ -1,63 +1,54 @@
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)
*/
Node mergeKList(Node[]a, int N)
{
/**
* This function merge K sorted LinkedList
*
* @param a array of LinkedList
* @param N size of array
* @return node
*/
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();
min.add(head.next);
Node curr = head;
//merging LinkedList
while(!min.isEmpty()) {
// merging LinkedList
while (!min.isEmpty()) {
Node temp = min.poll();
curr.next = temp;
curr = temp;
//Add Node in min Heap only if temp.next is not null
if(temp.next != null){
// Add Node in min Heap only if temp.next is not null
if (temp.next != null) {
min.add(temp.next);
}
}
return head;
}
// Node Class
private class Node{
private int data;
private Node next;
public Node(int d)
{
this.data = d;
next = null;
}
return head;
}
}
private class Node {
private int data;
private Node next;
public Node(int d) {
this.data = d;
next = null;
}
}
}