关注
合并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;
}
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 在大厂上班是一种什么样的体验 #
4835次浏览 68人参与
# 找工作的破防时刻 #
251758次浏览 1953人参与
# 程序员找工作至少要刷多少题? #
9535次浏览 147人参与
# 你投递的公司有几家约面了? #
157831次浏览 995人参与
# 论秋招对个人心气的改变 #
6006次浏览 104人参与
# 我的AI电子员工 #
26164次浏览 159人参与
# OPPO求职进展汇总 #
771005次浏览 5396人参与
# 为了减少AI幻觉,你注入过哪些设定? #
1912次浏览 55人参与
# 刚入职的你踩过哪些坑 #
4283次浏览 91人参与
# 程序员能干到多少岁? #
5849次浏览 90人参与
# 一张图晒一下你的AI员工 #
3034次浏览 62人参与
# 牛客AI体验站 #
3457次浏览 91人参与
# 我现在比当时_,你想录用我吗 #
3723次浏览 63人参与
# 想辞职但是不敢的原因 #
20183次浏览 88人参与
# ai智能作图 #
624794次浏览 5627人参与
# AI Coding的使用心得 #
2385次浏览 60人参与
# 关于春招/暑期实习,你想知道哪些信息? #
4051次浏览 84人参与
# 牛客租房专区 #
146783次浏览 1464人参与
# 晒晒你司的新年福利 #
4058次浏览 70人参与
# 虾皮求职进展汇总 #
373477次浏览 2792人参与

字节跳动公司福利 1366人发布