两有序数组间相加的TOPK问题
给定两有序数组arr1和arr2,以及整数k,返回来自arr1和arr2两个数相加和最大的前k个,此外两数必须来自不同的数组。
示例:
arr1 = [1 2 3 4 5]
arr2 = [3 5 7 9 11]
k = 4
输出
[16 15 14 14]
要求时间复杂度为O(klog(k))
解决方案:
我们发现最大的元素为arr1[M - 1] 与 arr2[N - 1]之和,比起稍小的两组数据分别为arr1[M - 1] + arr2[N - 2]和arr1[M - 2] + arr2[N - 1]。
定义一大根堆,初始将arr1[M - 1] + arr2[N - 1]存入堆中。
然后使用bfs,每次从堆中弹出最大值元素记做(i, j),并将其的(i - 1, j) 和 (i , j - 1)入堆。如此弹出k个元素即可。
public class Main{
public static class Node{
int x;
int y;
public Node(int x, int y) {
this.x = x;
this.y = y;
}
}
public static int[][] directs = {{0, -1}, {-1, 0}};
public static List<Integer>selution(int[] arr1, int[] arr2, int k) {
Queue<Node> maxHeap = new PriorityQueue<>(new Comparator<Node>() {
public int compare(Node node1, Node node2) {
return (arr1[node2.x] + arr2[node2.y]) - (arr1[node1.x] + arr2[node1.y]);
}
});
List<Integer> ans = new ArrayList<>(k);
int M = arr1.length;
int N = arr2.length;
boolean[][] visted = new boolean[M][N];
maxHeap.add(new Node(M - 1, N - 1));
visted[M - 1][N - 1] = true;
for(int i = 0; i < k; i++) {
Node cur = maxHeap.remove();
ans.add(arr1[cur.x] + arr2[cur.y]);
for(int[] direct : directs) {
int x = cur.x + direct[0];
int y = cur.y + direct[1];
if(x < 0 || y < 0 || visted[x][y]) {
continue;
}
maxHeap.add(new Node(x, y));
}
}
return ans;
}
}
document.querySelectorAll('.github-emoji') .forEach(el => { if (!el.dataset.src) { return; } const img = document.createElement('img'); img.style = 'display:none !important;'; img.src = el.dataset.src; img.addEventListener('error', () => { img.remove(); el.style.color = 'inherit'; el.style.backgroundImage = 'none'; el.style.background = 'none'; }); img.addEventListener('load', () => { img.remove(); }); document.body.appendChild(img); });
- 域名“宝贝”baby.cn以71万元价格结拍
- 温故而知新:设计模式之组合模式(Composite)
- ruby学习笔记(7)-闭包
- ruby学习笔记(6)-Array的使用
- centos7下部署iptables环境纪录(关闭默认的firewalle)
- ruby学习笔记(5)-模块module的运用
- linux系统root密码遗忘的情况下的解决办法
- ruby学习笔记(4)-动态修改类的属性
- 如果技术是一种生命
- ruby学习笔记(2)--类的基本使用
- 域名资讯:四声域名BHHS.com被BHHS公司收购
- ruby学习笔记(1)--初识语法
- 无法启用数据库中的 Service Broker,因为已存在启用的具有相同 ID 的 Service Broker。
- Centos7.2下针对LDAP的完整部署记录
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- [译] 在 Vue 组件中分离 UI 和业务逻辑
- Android Notes|BottomNavigationView 爱上 Lottie
- Android Notes|玩转 ShapeableImageView
- 前端技术 Webpack(学习 Webpack 的原因,Webpack 快速入门)
- 每日一问第1期 | 截取字符串
- 动态规划此一篇就够了 万字总结
- 每日一问第2期 | final, finally, finalize 的区别?
- 当return遇到try、catch、finally时会发生什么?
- 五分钟C语言数据结构 之 二叉树层次遍历
- 为何IntelliJ IDEA比Eclipse好在哪里?
- 五分钟C语言数据结构 之 二叉树中序遍历
- Django 安全之跨站点请求伪造(CSRF)保护
- 五分钟C语言数据结构 之 二叉树先序遍历
- Java 语言基础(常用设计原则和设计模式,常用 Java 8~11 新特性)
- 五分钟C语言数据结构 之 二叉树后序遍历(非递归很重要)