码迷,mamicode.com
首页 > 其他好文 > 详细

【Merge K Sorted Lists】cpp

时间:2015-05-18 20:46:25      阅读:105      评论:0      收藏:0      [点我收藏+]

标签:

题目:

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

代码:

/**
 * 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;
        return Solution::mergeK(lists, 0, lists.size()-1);
    }
    static ListNode* mergeK(vector<ListNode*>& lists, int begin, int end )
    {
            if ( begin==end ) return lists[begin];
            if ( (begin+1)==end ) return Solution::mergeTwo(lists[begin], lists[end]);
            int mid = ( begin + end ) / 2;
            ListNode *firstHalf = Solution::mergeK(lists, begin, mid);
            ListNode *secondHalf = Solution::mergeK(lists, mid+1, end);
            return Solution::mergeTwo(firstHalf, secondHalf);
    }
    static ListNode* mergeTwo( ListNode *h1, ListNode *h2 )
    {
            ListNode dummy(-1);
            ListNode *p = &dummy;
            while ( h1 && h2 ){
                if ( h1->val<h2->val ){
                    p->next = h1;
                    h1 = h1->next;
                }
                else{
                    p->next = h2;
                    h2 = h2->next;
                }
                p = p->next;
            }
            p->next = h1 ? h1 : h2;
            return dummy.next;
    }
};

tips:

多路归并写法。

【Merge K Sorted Lists】cpp

标签:

原文地址:http://www.cnblogs.com/xbf9xbf/p/4512861.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!