【算法速刷(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;
    }
};
相关推荐
半桔1 小时前
【STL源码剖析】二叉世界的平衡:从BST 到 AVL-tree 和 RB-tree 的插入逻辑
java·数据结构·c++·算法·set·map
塔中妖2 小时前
【华为OD】分割数组的最大差值
数据结构·算法·华为od
weixin_307779132 小时前
最小曲面问题的欧拉-拉格朗日方程 / 曲面极值问题的变分法推导
算法
RTC老炮2 小时前
webrtc弱网-AlrDetector类源码分析与算法原理
服务器·网络·算法·php·webrtc
孤廖2 小时前
【算法磨剑:用 C++ 思考的艺术・Dijkstra 实战】弱化版 vs 标准版模板,洛谷 P3371/P4779 双题精讲
java·开发语言·c++·程序人生·算法·贪心算法·启发式算法
sali-tec2 小时前
C# 基于halcon的视觉工作流-章33-矩状测量
开发语言·人工智能·算法·计算机视觉·c#
songx_993 小时前
leetcode29( 有效的括号)
java·数据结构·算法·leetcode
于樱花森上飞舞3 小时前
【java】常见排序算法详解
java·算法·排序算法
GawynKing3 小时前
图论3 图的遍历
算法·深度优先
东方芷兰5 小时前
Leetcode 刷题记录 21 —— 技巧
java·算法·leetcode·职场和发展·github·idea