23. Merge k Sorted Lists

来源:互联网 发布:建材价格查询软件 编辑:程序博客网 时间:2024/05/23 05:09

题目:Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
思路:将K个有序的链表合并成一个有序的链表,可以利用分而治之的方法去做,将lists中的链表分成两部分,先分别合并这两部分的链表,然后再将这两部分的链表进行合并,这样就将问题逐渐转化为两个有序链表的合并,代码如下:

/** * Definition for singly-linked list. * struct ListNode { *     int val; *     ListNode *next; *     ListNode(int x) : val(x), next(NULL) {} * }; */class Solution {public:    ListNode* getMergeLists(vector<ListNode*>& lists, int bgn, int end) {        if (bgn > end) return NULL;        else if (bgn == end) return lists[bgn];        else {            int mid = bgn + (end - bgn) / 2;            ListNode* left = getMergeLists(lists, bgn, mid);            ListNode* right = getMergeLists(lists, mid+1, end);            if (!left) return right;            else if (!right) return left;            ListNode *head = NULL, *pre = NULL;            while (left && right) {                if (left->val < right->val) {                    if (!head) {                        head = pre = left;                    }                    else {                        pre->next = left;                        pre = left;                    }                    left = left->next;                }                else {                    if (!head) {                        head = pre = right;                    }                    else {                        pre->next = right;                        pre = right;                    }                    right = right->next;                }            }            if (!left) pre->next = right;            else pre->next = left;            return head;        }    }    ListNode* mergeKLists(vector<ListNode*>& lists) {        return getMergeLists(lists, 0, lists.size()-1);    }};