4. 寻找两个正序数组的中位数
给定两个大小分别为 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
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 10^6
解题
方法:二分查找(变体)
两个数组长度确定,那我们就可以确认中位数的具体值,计作第k个
那我们分别从两个数组中找到前 k/2 个 数,谁的数小,就把谁排除,因为我们需要的是大的那个数
一直减下去,直到k=1,此时,对比两个数组的第一个数,谁小谁就是我们要找的数,
这个过程就是将前k-1个数给排除掉,然后剩下的第一个就是我们要找的数
因为有两个数组,这个过程要注意下标越界的问题,如果有一个数组越界了,说明这个数组的数全排除完了,
那就只看另外一个数组就好了
// 时间log(n+m),空间1
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.length;
int n = nums2.length;
// 这里有个小技巧,求中位数要根据奇偶数来找,有个公式,可以忽略奇偶数的问题
// 假设长度为 n,求中位数k = ((n+1)/2 + (n+2)/2) /2
int left = (n+m+1)/2;
int right = (n+m+2)/2;
// 找到第left 第right个数
return (findNum(nums1, 0, nums2, 0, left) + findNum(nums1, 0, nums2, 0, right)) / 2.0;
}
// i,j代表数组的下标, k代表找到第k个元素
private int findNum (int[] nums1, int i, int[] nums2, int j, int k) {
// nums1数组找完了,就找nums2数组中第k个数
if (i >= nums1.length) {
return nums2[j + k - 1];
}
if (j >= nums2.length) {
return nums1[i + k - 1];
}
if (k == 1) {
return nums1[i] < nums2[j]? nums1[i] : nums2[j];
}
// 分别从两个数组中找到前 k/2 个 数,谁的数小,就把谁排除
int num1 = (i + k/2 - 1) < nums1.length ? nums1[i + k/2 - 1] : Integer.MAX_VALUE;
int num2 = (j + k/2 - 1) < nums2.length ? nums2[j + k/2 - 1] : Integer.MAX_VALUE;
if (num1 > num2) {
// 将小的数组前k/2个排除掉,那下标就是从num1 / num2的下一位开始,找剩下第k-k/2个数
return findNum(nums1, i, nums2, j + k/2, k-k/2);
} else {
return findNum(nums1, i + k/2, nums2, j, k-k/2);
}
}
}