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

Leetcode merge-k-sorted-lists(合并k路有序链表 最小堆)

时间:2020-04-28 23:13:44      阅读:64      评论:0      收藏:0      [点我收藏+]

标签:链表   lin   复杂   合并   leetcode   nbsp   title   solution   wap   

题目描述

合并k个已排序的链表并将其作为一个已排序的链表返回。分析并描述其复杂度。 
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
 
思路:合并k路有序链表,可以用最小堆,然后每次送入堆中,取出最小的,并将该链表下一值取出放入
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    static bool cmp(const ListNode *l1, const ListNode *l2)
    {
        return l1->val<l2->val;
    }
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        int n = lists.size();
        ListNode *res = new ListNode(-1);
        if (n<1)
            return res->next;
        if (n<2)
            return lists[0];
        ListNode *p = res;
        int m = 0;
        for(int i=0;i<n;++i)
        {
            if(lists[i]==NULL)
            {
                swap(lists[m],lists[i]);
                m++;
            }
        }
        while(m < n-1)
        {
            sort(lists.begin()+m,lists.end(),cmp);
            p->next=lists[m];
            p=p->next;
            lists[m]=lists[m]->next;
            if(lists[m]==NULL)
            {
                m++;
            }
        }
        if(lists[m]!=NULL)
            p->next=lists[m];
        return res->next;
    }
};

 

Leetcode merge-k-sorted-lists(合并k路有序链表 最小堆)

标签:链表   lin   复杂   合并   leetcode   nbsp   title   solution   wap   

原文地址:https://www.cnblogs.com/zl1991/p/12797960.html

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