Leetcode由浅入深（三）Merge问题：合众为一

今天我们来看看Merge问题。

一、Merge Two Sorted Linked List

第一个问题是将两个排序过的LinkedList进行合并。 说到链表，我们就可以采用recursive的方法进行处理。

Java 代码如下

public ListNode mergeTwoLists(ListNode l1, ListNode l2){
		if(l1 == null) return l2;
		if(l2 == null) return l1;
		if(l1.val < l2.val){
			l1.next = mergeTwoLists(l1.next, l2);
			return l1;
		} else{
			l2.next = mergeTwoLists(l1, l2.next);
			return l2;
		}
}

比较两个表头元素，并使用递归遍历整个链表。

二、Merge Two Sorted Array

第二个问题是第一个问题的变形，如果提供的不是不是链表，而是两个排序过的数组，那么我们希望把两个数组合并到一个数组中，又该怎么处理呢？ 我们已知的是数组一有足够的长度，我们要把两个数组合并到数组一。

基本想法是从后向前从大到小的排序，这样可以将数组中值从大到小排列。

Java代码如下：

public void merge(int A[], int m, int B[], int n) {
    int i=m-1, j=n-1, k=m+n-1;
    while (i>-1 && j>-1) A[k--]= (A[i]>B[j]) ? A[i--] : B[j--];
    while (j>-1)         A[k--]=B[j--];
}

三、Merge K Linked List

问题三是问题一的另一种变化。当数量从2变为多，我们就要使用Heap这个数据结构来帮我们选出每个链表的最小值。

Java代码如下：

public class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        int k=lists.length;
        if(lists==null||k==0){
            return null;
        }
        PriorityQueue<ListNode> queue=new PriorityQueue<ListNode>(k,new Comparator<ListNode>(){
            @Override
            public int compare(ListNode o1,ListNode o2){
                if(o1.val<o2.val){
                    return -1;
                }
                else if(o1.val==o2.val){
                    return 0;
                }
                else{
                    return 1;
                }
            }
        });
        
        ListNode dummy=new ListNode(0);
        ListNode tail=dummy;
        for(ListNode node:lists){
            if(node!=null){
                queue.add(node);
            }
        }
        while(!queue.isEmpty()){
            tail.next=queue.poll();
            tail=tail.next;
            if(tail.next!=null){
                queue.add(tail.next);
            }
        }
        return dummy.next;
    }
}