加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
LC23.cpp 1.04 KB
一键复制 编辑 原始数据 按行查看 历史
xinanXu 提交于 2024-01-25 21:12 . 23. 合并 K 个升序链表
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeKLists(vector<ListNode*>& lists) {
int len = lists.size();
ListNode* ret = new ListNode(-1);
ListNode* ans = ret;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> orderedNodes;
for (int i = 0; i < len; i++) {
if (lists[i]) orderedNodes.push({lists[i]->val, i});
}
int idx = -1;
while (!orderedNodes.empty()) {
ret->next = new ListNode(orderedNodes.top().first);
ret = ret->next;
idx = orderedNodes.top().second;
orderedNodes.pop();
lists[idx] = lists[idx]->next;
if (lists[idx]) orderedNodes.push({lists[idx]->val, idx});
}
return ans->next;
}
};
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化