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
// 双指针 没有达到题目要求的时间复杂度
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.length;
int n = nums2.length;
int len = m+n;
int pre = -1, now = -1;
int numS1 = 0, numS2 = 0;
for(int i = 0; i< len/2+1; i++){
pre = now;
if(numS1 < m && (numS2 >= n || nums1[numS1] < nums2[numS2])){
now = nums1[numS1 ++];
}else{
now = nums2[numS2 ++];
}
}
if(len % 2==1){
return now;
}else{
return (double)(pre+now)/2;
}
}
}