题解 | #合并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存储数据

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务