Median of Two Sorted Arrays

題目描述:

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

You may assume nums1 and nums2 cannot be both empty.

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

解題思路:

由於時間限制不能對nums1和nums2數組進行二分查找,所以想着對中位數的序號進行二分查找

中位數的序號(從1開始計數):(nums1.size()+nums2.size() +1)/2,(nums1.size()+nums2.size() +2)/2,如果nums1.size()+nums2.size()爲奇數,則兩個序號相等,此時兩個有序的數組剛好只需查找一箇中位數序號;如果nums1.size()+nums2.size()爲偶數, 則兩個序號剛好爲,平均求中位數的序號;

對序號進行二分查找,首先排除nums1和nums2數組中不可能成爲序號K的值。i表示nums1中沒有被排除掉的子序列的起始下標,j表示nums2中沒有被排除掉的子序列的起始下標;如果nums1[i+k/2-1] < nums1[j+k/2-1] 則表示nums1中從序號i開始的k/2個數不可能成爲序號k對應的數值,反之亦然,從而排除了k/2個不可能成爲序號k對應的數值,遞歸下去則可以選擇出序號k對應的數值。

具體代碼如下:

 class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int i=0;
        int j=0;
        int size1 = nums1.size();
        int size2 = nums2.size();
        int k1 = (size1+size2+1)/2;
        int k2 = (size1+size2+2)/2;
        return (findKthValue(nums1,0,nums2,0,k1) + findKthValue(nums1,0,nums2,0,k2)) / 2.0; 
    }
    double findKthValue(vector<int>& nums1,int i, vector<int>& nums2, int j, int k)
    {
        if(i >= nums1.size())
            return nums2[j+k-1];
        if(j >= nums2.size())
            return nums1[i+k-1];
        if(k == 1)
            return min(nums1[i],nums2[j]);
        double middleValue1 = i+k/2-1 < nums1.size() ? nums1[i+k/2-1] : (numeric_limits<double>::max)();
        double middleValue2 = j+k/2-1 < nums2.size() ? nums2[j+k/2-1] : (numeric_limits<double>::max)(); 
        if(middleValue1 < middleValue2)
            return findKthValue(nums1,i+k/2,nums2,j,k-k/2);
        else
            return findKthValue(nums1,i,nums2,j+k/2,k-k/2);
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章