关注
合并N个有序数组:
public static List<Integer> mergeNArrays(int n, List<List<Integer>> list) {
List<Integer> res = new ArrayList<>();
// 数组idx,值idx,值
PriorityQueue<int []> pq = new PriorityQueue<>((a, b) -> a[2] - b[2]);
for (int i = 0; i < n; i++) {
if (!list.get(i).isEmpty()) {
pq.offer(new int[]{i, 0, list.get(i).get(0)});
}
}
while (!pq.isEmpty()) {
int []smallest = pq.poll();
int arrayIdx = smallest[0];
int idx = smallest[1];
res.add(smallest[2]);
if (idx < list.get(arrayIdx).size() - 1) {
idx++;
pq.offer(new int[]{arrayIdx, idx, list.get(arrayIdx).get(idx)});
}
}
return res;
}
查看原帖
点赞 评论
相关推荐
蛀虫害虫:强的我害怕 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 什么是优秀的实习经历 #
8388次浏览 211人参与
# 实习简历求拷打 #
12237次浏览 157人参与
# 被上班搭子“传染”了哪些习惯 #
5551次浏览 99人参与
# 秋招被挂春招仍然能投的公司 #
6790次浏览 99人参与
# 工作后,你落下了哪些病根 #
13423次浏览 187人参与
# mt对你说过最有启发的一句话 #
35798次浏览 427人参与
# 外包能不能当跳板? #
54089次浏览 256人参与
# 作业帮求职进展汇总 #
82998次浏览 547人参与
# 摸鱼被leader发现了怎么办 #
101271次浏览 642人参与
# 秋招特别不鸣谢 #
15685次浏览 177人参与
# 考研失败就一定是坏事吗? #
201020次浏览 1371人参与
# 选实习,你更看重哪方面? #
14051次浏览 216人参与
# 投格力的你,拿到offer了吗? #
152524次浏览 817人参与
# 今年秋招你收到了多少封邮件? #
18011次浏览 219人参与
# 京东美团大战,你怎么看? #
158107次浏览 860人参与
# 机械/制造每日一题 #
80251次浏览 1411人参与
# 担心入职之后被发现很菜怎么办 #
266272次浏览 1133人参与
# 第一次面试 #
1036481次浏览 13683人参与
# 携程求职进展汇总 #
840105次浏览 5532人参与
# 毕业论文进行时 #
20892次浏览 131人参与