标签:EDA art 要求 tar pre turn 就是 == arrays
传送门:[LeetCode] 4. 寻找两个有序数组的中位数
给定两个大小为 m 和 n 的有序数组 nums1
和 nums2
。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1
和 nums2
不会同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5
k == 1
,或者有一个数组为空。k == 1
结束,返回两个数组的第一个数中较小值;若有一个数组为空,返回另一个数组中的第 k 个数。class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int n = nums1.length;
int m = nums2.length;
if (((m + n) & 1) == 1) {
return getKth(nums1, 0, nums2, 0, (n + m + 1) / 2);
} else {
return (getKth(nums1, 0, nums2, 0, (n + m + 1) / 2)
+ getKth(nums1, 0, nums2, 0, (n + m + 2) / 2)) / 2.0;
}
}
public int getKth(int[] nums1, int start1, int[] nums2, int start2, int k) {
int len1 = nums1.length - start1;
int len2 = nums2.length - start2;
if (len1 == 0) {
return nums2[start2 + k - 1];
}
if (len2 == 0) {
return nums1[start1 + k - 1];
}
if (k == 1) {
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]) {
return getKth(nums1, start1, nums2, j + 1, k - (j - start2 + 1));
} else {
return getKth(nums1, i + 1, nums2, start2, k - (i - start1 + 1));
}
}
}
这题的二分是对 k 进行二分,每次能排除 k/2 个数。
在一个数组里二分简单,在这题里学习在两个数组里进行二分。
标签:EDA art 要求 tar pre turn 就是 == arrays
原文地址:https://www.cnblogs.com/qiu_jiaqi/p/LeetCode_4.html