题目链接
给定两个大小为 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
题解:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by-w-2/
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m=nums1.length;
int n=nums2.length;
if(m>n){
return findMedianSortedArrays(nums2,nums1);
}
if(m==0){
if(n%2==1) return nums2[n/2];
else return (nums2[n/2]+nums2[n/2-1])*0.5;
}
int l=0,r=m;
int mx=0,mn=0;
while(l<=r){
int i=(l+r)>>1;
int j=(m+n+1)/2-i;
if(i==0) mx=nums2[j-1];
else if(j==0) mx=nums1[i-1];
else mx=Math.max(nums1[i-1],nums2[j-1]);
if(i==m) mn=nums2[j];
else if(j==n) mn=nums1[i];
else mn=Math.min(nums1[i],nums2[j]);
if(mx<=mn) return (m+n)%2==1?mx:(mx+mn)*0.5;
else{
if(i==0) l=i+1;
else if(j==0) r=i-1;
else{
if(nums1[i-1]>=nums2[j-1]) r=i-1;
else l=i+1;
}
}
}
return 0.0;
}
}