【算法速刷(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;
    }
};
相关推荐
Voyager_410 分钟前
图像处理踩坑:浮点数误差导致的缩放尺寸异常与解决办法
数据结构·图像处理·人工智能·python·算法
文艺倾年16 分钟前
【八股消消乐】手撕分布式协议和算法(基础篇)
分布式·算法
万岳科技系统开发1 小时前
从源码优化外卖配送系统:算法调度、智能推荐与数据分析应用
算法·数据挖掘·数据分析
信奥卷王4 小时前
[GESP202503 五级] 原根判断
java·数据结构·算法
兮山与4 小时前
算法4.0
算法
nju_spy4 小时前
力扣每日一题(二)任务安排问题 + 区间变换问题 + 排列组合数学推式子
算法·leetcode·二分查找·贪心·排列组合·容斥原理·最大堆
初听于你4 小时前
高频面试题解析:算法到数据库全攻略
数据库·算法
翟天保Steven4 小时前
ITK-基于Mattes互信息的二维多模态配准算法
算法
代码对我眨眼睛4 小时前
226. 翻转二叉树 LeetCode 热题 HOT 100
算法·leetcode·职场和发展
黑色的山岗在沉睡5 小时前
LeetCode 494. 目标和
算法·leetcode·职场和发展