【算法速刷(10/100)】LeetCode —— 23. 合并 K 个升序链表

按照最朴素的方法,每轮都对所给列表进行一次遍历,O(n)的复杂度获得值最小的节点,并将其上的链表指针后移一位,一旦为空则剔除数组。数组为空时结束循环。

这样写时间复杂度较高,因为涉及到枚举最小值节点,数组的删除操作,但空间复杂度可以降到O(1)

cpp 复制代码
class Solution {
public:
    ListNode* mergeKLists(vector<ListNode*>& lists) {
        
        int n = lists.size();

        for(int i = n - 1; i >= 0; i--)
        {
            if(lists[i] == nullptr)
                lists.erase(lists.begin() + i);
        }

        n = lists.size();
        if(n == 0)
            return nullptr;

        ListNode* head = nullptr, *p = nullptr;
        while(lists.size() > 0)
        {
            n = lists.size();

            int minId = 0;
            for(int i = 1; i < n; i++)
            {
                if(lists[i]->val < lists[minId]->val)
                {
                    minId = i;
                }
            }

            if(!head)
            {
                head = p = lists[minId];
            }
            else
            {
                p->next = lists[minId];
                p = p->next;
            }

            lists[minId] = lists[minId]->next;
            if(!lists[minId])
                lists.erase(lists.begin() + minId);
        }

        return head;
    }
};
相关推荐
夏鹏今天学习了吗31 分钟前
【LeetCode热题100(47/100)】路径总和 III
算法·leetcode·职场和发展
smj2302_7968265236 分钟前
解决leetcode第3721题最长平衡子数组II
python·算法·leetcode
m0_626535201 小时前
力扣题目练习 换水问题
python·算法·leetcode
第六五1 小时前
DPC和DPC-KNN算法
人工智能·算法·机器学习
一匹电信狗1 小时前
【LeetCode_160】相交链表
c语言·开发语言·数据结构·c++·算法·leetcode·stl
再卷也是菜2 小时前
C++篇(14)二叉树进阶算法题
c++·算法
小邓儿◑.◑2 小时前
贪心算法 | 每周8题(三)
算法·贪心算法
2401_841495642 小时前
【数据结构】最长的最短路径的求解
java·数据结构·c++·python·算法·最短路径·图搜索
小龙报2 小时前
《算法每日一题(1)--- 连续因子》
c语言·开发语言·c++·windows·git·算法·visual studio
夜晚中的人海2 小时前
【C++】滑动窗口算法习题
开发语言·c++·算法