4. 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)).

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

思路

看到O(log (m+n))，很容易就联想到二分查找。也正因为这样，被二分查找的寻找中间的数所困扰，浪费很多时间。其实，这道题可以把寻找中位数，改成寻找第K小的数。算法的关键就是找对划分的标签。如果按照二分查找的做法，寻找mid为标签，将会很复杂。这道题以k/2为标签。用A[k/2]和B[k/2]作比较，如果A的较小，则抛弃A的前k/2,反之抛弃B的前k/2。如果两者相等，就找到了第k小的数了。

代码块

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {

        if ((nums1.size()+nums2.size())%2!=0)
            return findK(nums1,nums2,(nums1.size()+nums2.size())/ 2 + 1);
        else
            return (findK(nums1,nums2,(nums1.size()+nums2.size())/ 2)+findK(nums1,nums2,(nums1.size()+nums2.size())/ 2 + 1)) / 2.00;
    }

    float findK(vector<int> nums1, vector<int> nums2,int k)
    {
        if (nums1.size() < nums2.size()) {
            return findK(nums2,nums1,k);//保证nums1始终最大
        }
        if (nums2.size() == 0)
            return nums1[k - 1];
        if (k == 1) {
            return min(nums1[0],nums2[0]);
        }
        int p2 = min(k / 2, nums2.size()), p1 = k - p2;

        if (nums1[p1 - 1] < nums2[p2 - 1]) {
            nums1.erase(nums1.begin(),nums1.begin()+p1);
            return findK(nums1,nums2,k - p1);
        }
        else if (nums1[p1 - 1] > nums2[p2 - 1]) {
            nums2.erase(nums2.begin(),nums2.begin()+p2);
            return findK(nums1,nums2, k - p2);
        }
        else
            return nums1[p1 - 1];
    }

    int min(int a,int b) {
        if(a>b) {
            return b;
        }   
        else {
            return a;
        }
    }
};