首页 > 其他 > 详细

[Leetcode]-- Merge k Sorted Lists

时间:2014-01-28 20:59:54      阅读:446      评论:0      收藏:0      [点我收藏+]

Solution and Precautions:

Very similar to the classical merger sort for two sorted arrays, since the k linked list are already sorted, we just find the smallest number among the heads list[i][0] (i=0,…k-1), of course, if some list already reached to the end, that is list[i][0] is NULL, we just ignore this list, the smallest number is obviously the smallest one in the combined or merged final list, we repeat this process and push the smallest number found into the tail of the final merger list until there is no elements leaft. To fast find the smallest number among the k head elements, we could adopt the heap data structure, thus each found and delete of the minHeap, it cost O(log k), and it is possible to perform n such operations where n is the total number of elements of the k list, thus the time complexity is bounded by O(n * log k).

Note, the bound could be even tighter since the size of the heap could be smaller when some list is consumed up. So for the delete min operations later on it might cost less than log k. (The exact bound is not easy to figure out for me)

 

Min Heap Method

bubuko.com,布布扣
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
 /* Min heap*/
public class Solution {
    public ListNode mergeKLists(ArrayList<ListNode> lists) {
       if(lists == null || lists.size() == 0) return null;
       
       Comparator<ListNode> comparator = new Comparator<ListNode>(){
           public int compare(ListNode m, ListNode n){
               if(m.val == n. val) return 0;
               else if(m.val > n .val) return 1;
               else return -1;
           }
       };
       
       
       PriorityQueue<ListNode> p = new PriorityQueue<ListNode>(lists.size(), comparator);
       
       for( int i = 0; i < lists.size(); i++){
           if(lists.get(i) != null)
            p.add(lists.get(i));
       }
       
       ListNode head = null, c =null;
       
       while(!p.isEmpty()){
           if(head == null) {
               head = p.poll();
               c = head;
           } else{
               c.next = p.poll();
               c = c.next;
           }
           
           if(c.next != null) p.add(c.next);
       }
       return head;
    }
    
}
bubuko.com,布布扣

 

ref:http://blog.csdn.net/zyfo2/article/details/8682727

 

 

 

 

 

Use Merge Two sorted List Method

bubuko.com,布布扣
 /* Merge Two lists method*/
public class Solution {
    public ListNode mergeKLists(ArrayList<ListNode> lists) {
        ListNode head = null;
        int len = lists.size();
        
        if(len == 0) 
            return null;
        else if(len ==1) 
            return lists.get(0);
        
        head = merge2List(lists.get(0), lists.get(1));
        for(int i =2; i< len; i++){
            head = merge2List(lists.get(i), head);
        }
        return head;
        
    }
    
    public ListNode merge2List(ListNode node1, ListNode node2){
        ListNode head = new ListNode(Integer.MIN_VALUE);
        ListNode tmp = head;
        while(node1 != null && node2 != null){
            if(node1.val <= node2.val){
                ListNode node = new ListNode(node1.val);
                tmp.next = node;
                tmp = tmp.next;
                node1 = node1.next;
            }else{
                 ListNode node = new ListNode(node2.val);
                tmp.next = node;
                tmp = tmp.next;
                node2 = node2.next;
            }
        }
        
        if(node1 != null){
            tmp.next = node1;
        }
        
        if(node2 != null){
            tmp.next = node2;
        }
        
        return head.next;
    }
    
}
bubuko.com,布布扣

[Leetcode]-- Merge k Sorted Lists

原文:http://www.cnblogs.com/RazerLu/p/3535608.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!