leetcode刷题23. 合并K个排序链表

题目描述:合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-k-sorted-lists
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:
1、将所有链表中的值放入vector容器中;
2、使用sort()对vector中的值进行排序;
3、将排序后的vector中的数保存到新的链表中,返回链表head指针
空间复杂度O(N),时间复杂度O(N)

/**
 * 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 nullptr;
        }
        if (lists.size() == 1) {
            return lists[0];
        }
        vector<int> res_list;
        // 将所有链表中的值放入vector中
        for(int i=0;i<lists.size();i++){
            ListNode* now = lists[i];
            while(now){
                res_list.push_back(now->val);
                now = now->next;
            }
        }
        sort(res_list.begin(),res_list.end()); // 排序
        // 将vector中的值放入新创建的链表中
        ListNode* cur = new ListNode(NULL);
        ListNode* head = cur;
        for(int i = 0;i < res_list.size();i++)
        {
            cur->val = res_list[i];
            if(i < res_list.size()-1)
                cur->next = new ListNode(NULL);
                cur = cur->next;
        }
        return head;

    }
};
发布了29 篇原创文章 · 获赞 0 · 访问量 502

猜你喜欢

转载自blog.csdn.net/weixin_43022263/article/details/104333529