编程题一:两有序数组相加后的前k个数组
描述 :给定数组如[1,2,3,4,5,6,7]与[5,6,7,8,9,10,11,12]。那么得到最大的前5个为[19,18,18,17,17]。
解题方法:二分+单指针
leetcode相似题目:668. 乘法表中第k小的数 - 力扣、2040. 两个有序数组的第 K 小乘积 - 力扣
public class Solution {
ArrayList<Integer> res = new ArrayList<>();
public int[] LIS(int[] arr1, int[] arr2, int k) {
int len = arr1.length;
int left = arr1[len - k] + arr2[len - k];
int right = arr1[len - 1] + arr2[len - 1];
//使用二分找到要求的最终的数组中最小的一个数字
while (left < right) {
int mid = (left + right) / 2;
int count = getCnt(arr1, arr2, mid, k);
int m = Math.min(len, k);
if (m * m - count <= k) right = mid;
else left = mid + 1;
}
//得到的left就是目标序列中最小的一个
getRes(arr1, arr2, left, k);
Collections.sort(res);
int[] ret = new int[k];
for (int i = res.size() - 1; i >= res.size() - k; i--) {
ret[res.size() - i - 1] = res.get(i);
}
return ret;
}
//使用和 getCnt相同的方法将结果数组加入到res中。
public void getRes(int[] arr1, int[] arr2, int left, int k) {
int len = arr1.length;
int r = len - 1;
res = new ArrayList<>();
for (int i = len - k; i < len; i++) {
while (arr1[i] + arr2[r] > left) r--;
for (int j = r; j < len && (arr1[i] + arr2[r]) >= left; j++) res.add(arr1[i] + arr2[j]);
}
}
//使用单个指针,根据两数组相加的表格的特性(从某行或者某列往后遍历必须是单调递增) 往前就是单调递减
//所以可以使用一个指针移动,可以将扫描表格的复杂度从k^2降低为k
//return 小于等于mid的数字的数量
public int getCnt(int[] arr1, int[] arr2, int mid, int k) {
int len = arr1.length;
int l = len - k, r = len - 1;
int cnt = 0;
for (int i = len - k; i < len; i++) {
while (arr1[i] + arr2[r] > mid) r--;
cnt += r - l + 1;
}
return cnt;
}
}
编程题二:最长递增子序列(进化)
描述:在最长递增子序列的基础上返回字典序最小的其中一个子序列。
牛客题目链接:最长递增子序列__牛客网
解题方法:贪心+二分
public class Solution {
public int[] LIS(int[] arr) {
int[] maxLen = new int[arr.length];
ArrayList<Integer> res = new ArrayList<>();
maxLen[0] = 1;
res.add(arr[0]);
for (int i = 0; i < arr.length; i++) {
if (arr[i] > res.get(res.size() - 1)) {
res.add(arr[i]);
maxLen[i] = res.size();
}
//二分法找到res中"最小的"比arr[i]大的数
int pos = lowBounce(res, arr[i]);
res.set(pos, arr[i]);
maxLen[i] = pos + 1;
}
//至此为止res是不准确的,只有res的长度是准确的。
//maxLen是对应arr的,是准确的。
//所以现在要根据maxLen得到准确的结果
for (int i = maxLen.length - 1, j = res.size(); i >= 0; i--) {
//在arr[i]中,最后的一个maxLen对应的就是最小的。
if (maxLen[i] == j) {
res.set(--j, arr[i]);
}
}
int[] ress = new int[res.size()];
for (int i = 0; i < res.size(); i++) {
ress[i] = res.get(i);
}
return ress;
}
public int lowBounce(ArrayList<Integer> res, int target) {
int l = 0, r = res.size() - 1;
while (l < r) {
int mid = (l + r) / 2;
if (res.get(mid) >= target) r = mid;
else l = mid + 1;
}
return l;
}
}
4299

被折叠的 条评论
为什么被折叠?



