[链表]合并K个已排序的链表

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

问题

思路

由于代码形参是直接给了一个数组，因此可以把每个节点都当成一个链表，然后通过每两个节点合并，构成新的链表，再与下一个节点合并，就可以实现整体K个链表的合并。但是这种方法会超时，无法ac。因此采用从中间将数组一分为二，直到分成一个个单独节点。最后将单独节点连接起来，得到合并后的K个链表。

代码实现

/**
 * 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) {
        return divMerge(lists, 0, lists.size() - 1);
    }
    
    // 将数组对半拆成一个个元素，再合并
    ListNode* divMerge(vector<ListNode*> &lists, int left, int right)
    {
        if (left > right) return NULL;
        if (left == right) return lists[left];
        
        int mid = left + (right - left) / 2;
        return mergeTwoList(divMerge(lists, left, mid), divMerge(lists, mid + 1, right));
    }
    
    // 合并两个链表
    ListNode* mergeTwoList(ListNode* p1, ListNode* p2)
    {
        ListNode* ret = new ListNode(0);
        ListNode* cur = ret;
        while (p1 && p2)
        {
            if (p1->val < p2->val)
            {
                cur->next = p1;
                p1 = p1->next;
            }
            else
            {
                cur->next = p2;
                p2 = p2->next;
            }
            cur = cur->next;
        }
        if (p1)
        {
            cur->next = p1;
        }
        if (p2)
        {
            cur->next = p2;
        }
        
        return ret->next;
    }
};