题目
给定两个大小为 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
解题思路
双指针法,每次比较两个数组指定位置的大小并移动指针位置
代码
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int nums1Length = nums1.length;
int nums2Length = nums2.length;
//移动的次数,偶数减1
int centre = (nums1Length + nums2Length) % 2 == 0 ? (nums1Length + nums2Length) / 2 - 1 : (nums1Length + nums2Length) / 2;
int nums1Left = 0;
int nums2Left = 0;
while(centre > 0) {
int value1 = nums1Left == nums1Length ? Integer.MAX_VALUE : nums1[nums1Left];
int value2 = nums2Left == nums2Length ? Integer.MAX_VALUE : nums2[nums2Left];
if (value1 < value2) {
nums1Left++;
} else {
nums2Left++;
}
centre--;
}
int first = (nums1Left == nums1Length || (nums2Left != nums2Length && nums1[nums1Left] > nums2[nums2Left]))
? nums2[nums2Left] : nums1[nums1Left];
//两个数组长度和为偶数时
if ((nums1Length + nums2Length) % 2 == 0) {
//如果nums1Left没有走到最后,并且first值等于nums1[nums1Left],表示first是在nums1数组
if (nums1Left != nums1Length && first == nums1[nums1Left]) {
nums1Left++;
} else {
nums2Left++;
}
int second = (nums1Left == nums1Length || (nums2Left != nums2Length && nums1[nums1Left] > nums2[nums2Left]))
? nums2[nums2Left] : nums1[nums1Left];
return (first + second) / 2.0;
}
return (double)first;
}