leetcode 数组Ⅱ 2021-02-18
寻找两个正序数组的中位数(困难)1
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。
进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗?
示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
注意:要想达到时间复杂度为 O(log (m+n)),这应该使用到二分查找法
//时间复杂度为O(m+n)
public double findMedianSortedArraysA(int[] nums1, int[] nums2) {
int len1 = nums1.length;
int len2 = nums2.length;
int[] nums = new int[len1 + len2];
int i = 0, j = 0, n = 0;
while (i < len1 && j < len2) { //如果有一个数组长度为0都不进行排序
if (nums1[i] < nums2[j]) {
nums[n] = nums1[i];
i++;
} else {
nums[n] = nums2[j];
j++;
}
n++;
}
for (; j < len2; j++) {
nums[n] = nums2[j];
n++;
}
for (; i < len1; i++) {
nums[n] = nums1[i];
n++;
}
if (nums.length % 2 == 0) {
return (float) (nums[nums.length / 2 - 1] + nums[nums.length / 2]) / 2;
} else {
return (float) nums[nums.length / 2];
}
}
二分查找法:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xun-zhao-liang-ge-you-xu-shu-zu-de-zhong-wei-s-114/
最佳观光组合(中等)1
给定正整数数组 A,A[i] 表示第 i 个观光景点的评分,并且两个景点 i 和 j 之间的距离为 j - i。
一对景点(i < j)组成的观光组合的得分为(A[i] + A[j] + i - j):景点的评分之和减去它们两者之间的距离。
返回一对观光景点能取得的最高分。
示例:
输入:[8,1,5,2,6]
输出:11
解释:i = 0, j = 2, A[i] + A[j] + i - j = 8 + 5 + 0 - 2 = 11
暴力循环,超时
public int maxScoreSightseeingPair(int[] A) {
int max = 0;
for (int i = 0;i < A.length; i ++){
for (int j = i+1;j < A.length; j ++){
if(A[i] + A[j] - (j - i) > max){
max = A[i] + A[j] - (j - i);
}
}
}
return max;
}
没想出来怎么优化,只能在评论里需求思路。
思路:不妨将 A[i] + A[j] + i - j 变成 (A[i] + i) + (A[j] - j)
在一次for循环中,当j的值一定,而i变化的时候,right的值是不变的
而left的值会随i变化而变化,我们只需取得left的最大值,即可得到这次j不变i变化的for循环的最大值。
也就是当 j 景点只与小于 j 的那些 i 进行 (A[i] + i) + (A[j] - j) 取分数,那当A[i] + i最大时,则(A[i] + i) + (A[j] - j)一定是最大的。
public int maxScoreSightseeingPair(int[] A) {
int max = Integer.MIN_VALUE,left = A[0]; // left初始为A[0] + 0
for (int j = 1;j < A.length; j ++){
if(A[j] - j + left > max){
max = A[j] - j + left;
}
if(A[j] + j > left){
left = A[j] + j;
}
}
return max;
}
数组标签完成度