【算法速刷(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;
    }
};
相关推荐
UnderTheTime20 分钟前
2025 XYD Summer Camp 7.10 筛法
算法
zstar-_20 分钟前
Claude code在Windows上的配置流程
笔记·算法·leetcode
圆头猫爹35 分钟前
第34次CCF-CSP认证第4题,货物调度
c++·算法·动态规划
秋说38 分钟前
【PTA数据结构 | C语言版】出栈序列的合法性
c语言·数据结构·算法
用户40315986396631 小时前
多窗口事件分发系统
java·算法
用户40315986396631 小时前
ARP 缓存与报文转发模拟
java·算法
hi0_61 小时前
03 数组 VS 链表
java·数据结构·c++·笔记·算法·链表
aPurpleBerry1 小时前
hot100 hot75 栈、队列题目思路
javascript·算法
卷福同学3 小时前
【AI编程】AI+高德MCP不到10分钟搞定上海三日游
人工智能·算法·程序员
mit6.8243 小时前
[Leetcode] 预处理 | 多叉树bfs | 格雷编码 | static_cast | 矩阵对角线
算法