23. 合并K个排序链表（java）

23. 合并K个排序链表

题目描述

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6

---

解法一：计数法

• 将所有的链表全部暂存在数组来计数
• 通过计数情况全部有序放进新创建的链表

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        int[] arr = new int[10000];
        for(ListNode list:lists){
            while(list!=null){
                arr[list.val+5000]++;
                list=list.next;
            }
        }
        ListNode link =new ListNode(0);
        ListNode ans=link;
        for(int i=0;i<arr.length;i++){
            int num=arr[i];
            while(num>0){
                ListNode temp =new ListNode(i-5000);
                link.next=temp;
                link=temp;
                num--;
            }
        }
        return ans.next;
    }
}

---

解法二：优先队列

步骤：

1. 创建（PriorityQueue）优先级队列，（写一个比较关系）
2. 将所有链表节点全部存进堆
3. 队列不为空，将链表首节点连接新表，若后面还有节点，则将除首节点外其他节点放进队列中

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if(lists==null||lists.length==0) return null;
        //由于是ListNode类型，写一个比较关系
        PriorityQueue<ListNode> queue = new PriorityQueue<>((a,b)->a.val-b.val);
        for(ListNode list:lists){
            //不为空才加入
            if(list!=null){
                queue.add(list);
            }
        }
        ListNode ans = new ListNode(0);
        ListNode link = ans;
        while(!queue.isEmpty()){
            link.next=queue.poll();
            link=link.next;
            if(link.next!=null) queue.add(link.next);
        }
        return ans.next;
    }
}

---

解法三：分而治之

步骤：

1.先判断基础条件

2.进入合并判断器，选择两两链表进入合并函数（当left==right时返回选择该链表），

3.对两条链表有序递归合并。

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if(lists==null||lists.length==0) return null;
        return merge(lists,0,lists.length-1);
    }
    public ListNode merge(ListNode[] lists,int left,int right){
        if(left==right) return lists[left];
        int mid = left+(right-left)/2;
        ListNode l1 = merge(lists,left,mid);
        ListNode l2 = merge(lists,mid+1,right);
        return mergeTowLists(lists,l1,l2);
    }
    public ListNode mergeTowLists(ListNode[] lists,ListNode l1,ListNode l2){
        if(l1==null) return l2;
        if(l2==null) return l1;
        if(l1.val<l2.val){
            l1.next=mergeTowLists(lists,l1.next,l2);
            //return紧跟其后，即若l1首结点小，则合并后l1为首结点
            return l1;
        }
        else{
            l2.next=mergeTowLists(lists,l1,l2.next);
            //反之l2为首节点
            return l2;
        }
    }
}