尋找兩個正序數組的中位數(困難)

題目鏈接
給定兩個大小爲 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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章