学习目标:
力扣第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
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
示例 4:
输入:nums1 = [], nums2 = [1]
输出:1.00000
示例 5:
输入:nums1 = [2], nums2 = []
输出:2.00000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
学习时间:
2020年10月25日
学习产出:
/*
* 因为两个数组都是有序的,寻找的是两个数组的中位数,因此我们每个数组只需要找一半就可以了
* 用双指针,通过循环将两个数组进行比较,较小的数给r,一直更新r和l,
* 当两个数组的和长度为奇数,就返回r,若为偶数,则返回r上一次更新的数,即l和r和的二分之一即可
*/
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.length;
int n = nums2.length;
int length = m + n;
int A = 0, B = 0;
int l = r = -1;
for (int i = 0; i <= length/2; i++) {
l = r;
if (A < m && (B >= n || nums1[A] < nums2[B])) {
r = nums1[A++];
} else {
r = nums2[B++];
}
}
if ((m + n) % 2 == 0) {
return (r + l) / 2.0;
}
return r;
}
这里贴一下进阶问题的解答(跪拜力扣大佬们),思路和逻辑不是一般的清晰
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int n = nums1.length;
int m = nums2.length;
int left = (n + m + 1) / 2;
int right = (n + m + 2) / 2;
return (getK(nums1, 0, n - 1, nums2, 0, m - 1, left) + getK(nums1, 0, n - 1, nums2, 0, m - 1, right))*0.5;
}
private int getK(int[] nums1, int start1, int end1, int[] nums2, int start2, int end2, int k) {
int len1 = end1 - start1 + 1;
int len2 = end2 - start2 + 1;
if (len1 > len2)//保证数组有空的话一定是nums1,方便后面数组为0的时候返回值
return getK(nums2, start2, end2, nums1, start1, end1, k);
if (len1 == 0)
return nums2[start2 + k - 1];
if (k == 1)//当k==1即为下一个数就是我们要找的中位数,则中位数是nums1和nums2中较小的那个
return Math.min(nums1[start1], nums2[start2]);
int i = start1 + Math.min(len1, k / 2) - 1;
int j = start2 + Math.min(len2, k / 2) - 1;
if (nums1[i] > nums2[j]) {
//如果nums1[i]大于nums2[j],则nums2数组的开始变为了j+1,k变成了k-(j-开始的下标+1(下标要加一))
return getK(nums1, start1, end1, nums2, j + 1, end2, k - (j - start2 + 1));
} else {
return getK(nums1, i + 1, end1, nums2, start2, end2, k - (i - start1 + 1));
}
}