LeetCode 23 合并K个排序链表

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

示例:

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

思路

两两比较就行了 遍历k个排序链表 然后多次调用合并两个有序链表的方法

c++代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        ListNode *res;
        if(lists.size() <= 0)
            return NULL;
        else if(lists.size() == 1)
            return lists[0];
        else {
           res = lists[0];
           for(int i = 1; i < lists.size(); i++) {
               res = mergeTwoLists(res, lists[i]);
           } 
        }
        return res;
    }
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if(l1 == NULL && l2 == NULL)
            return NULL;
        if(l1 != NULL && l2 == NULL)
            return l1;
        if(l2 != NULL && l1 == NULL)
            return l2;
        ListNode* l,*res;
        if(l1->val > l2->val) {
            l = l2;
            l2 = l2->next;
        }     
        else {
            l = l1;
            l1 = l1->next;
        }    
        res = l;
        while(l1 && l2) {
            if(l1->val < l2->val) {
                l->next = l1;
                l = l1;
                l1 = l1->next;  
            }
            else {
                l->next = l2;
                l = l2;
                l2 = l2->next;
            }
        }
        if(l1) 
            l->next = l1;
        else
            l->next = l2;

        return res;
    }
};