Merge k sorted List

问题来源: leetcode

问题描述

合并k条有序链表。

问题分析

实际上，我们只需要把这个问题简化为n次两条有序链表的合并，一切就会迎刃而解。真正困难的部分也就是链表归并的代码。两路有序链表归并的思路也不复杂，就是不断的将两条链表中小的元素取出（按照升序）组成一条新的链表。

解决代码

/** * 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) {        if (lists.size() == 0) return NULL;        while (lists.size() > 1) {            lists.push_back(mergeList(*(lists.begin()), *((lists.begin()+1))));            lists.erase(lists.begin());            lists.erase(lists.begin());                }        return lists[0];    }    ListNode* mergeList(ListNode* l1, ListNode* l2) {        ListNode* tmp = NULL, *tmph = NULL;        while (l1 && l2) {            if (l1->val < l2->val) {                if (!tmp) tmph = tmp = l1;                else {                    tmp->next = l1;                    tmp = tmp->next;                }                l1 = l1->next;            } else {                if (!tmp) tmph = tmp = l2;                else {                    tmp->next = l2;                    tmp = tmp->next;                }                l2 = l2->next;            }        }        if (tmp)          tmp->next = l1 ? l1 : l2;        else          tmph = l1 ? l1 : l2;        return tmph;    }};