题解 | #合并k个已排序的链表#

合并k个已排序的链表

http://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6

sort函数进行排序

首先用vector存放所有链表指针,然后sort函数进行排序,注意自写比较函数,最后连接vector中的指针即可。

C++代码:

class Solution {
public:
    ListNode *mergeKLists(vector<ListNode *> &lists) {
        if (lists.empty()) {
            return NULL;
        }
        vector<ListNode *> ans;
        for (int i = 0; i < (int)lists.size(); i++) {
            ListNode *cur = lists[i];
            while (cur) {
                ans.push_back(cur);
                cur = cur->next;
            }
        }
        sort(ans.begin(), ans.end(), cmp);
        for (int i = 0; i <= (int)ans.size() - 2; i++) {
            ans[i]->next = ans[i + 1];
        }
        ans[ans.size() - 1]->next = NULL;
        return ans[0];
    }
    static bool cmp(ListNode *a, ListNode *b) {
        return a->val < b->val;
    }
};

时间复杂度O(nlogn):  两次遍历O(n),排序O(nlogn)

空间复杂度O(n):vector存储数据

全部评论

相关推荐

小浪_Coding:找硬件测试,也可兼顾软测欧, 简历还可以的 ,注意排版,项目写的有条理一点, 然后个人技能多加点, 润色好简历之后就开始沟通海投了,深圳,东莞这边做硬件相关的公司还不少, 医疗类,仪器类的都可以尝试
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务