【LeetCode】4. Median of Two Sorted Arrays两个有序数组的中位数

4. Median of Two Sorted Arrays

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

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

Example 1:

nums1= [1, 3]

nums2= [2]

 

Themedian is 2.0

Example 2:

nums1= [1, 2]

nums2= [3, 4]

 

Themedian is (2 + 3)/2 = 2.5

 

//分析题目，求两个数组合并的中位数。So，最简单的方法就是把两个数组合并，然后排序后取中位数。

但是！！！题意要求：The overall run time complexityshould be O(log (m+n)).

所以我们需要用更优的算法，这也是本题的难点！

通过题意可以发现，给定的两个数组是排序好的，所以：我们通过递归解决

int findkth(int a[],int m,int b,int n,int k){

//参数：a,m,b,n分别表示数组及其长度！k表示寻找排第k位的数

//递归边界：

为了便于后续计算，我们希望a的长度小于b，所以

if(m>n){

   return findkth(b,n,a,m,k);

  }

//如此，当m为0时（这是n不能为0因为m<n）

if(m==0){

   return b[k-1];

   }

//当k为1时

if(k==1){

  returnmin(a[0],b[0]);

  }

//以上是递归边界的建立，递归先设置好边界！

int i=min(m,k/2),j=k-i;

//从两个数组的数字进行比较

if(a[i-1]>b[j-1]){

   //如果a的大于b的，所以我们只需要递归比较a与b的后j个数组，寻找第k-j大的数字

  return findkth(a,m,b+j,n-j,k-j);

  }else{

   returnfindkth(a+i,m-I,b,n,k-i);

   }

  return 0;

}

以上为递归函数 的构建！

class Solution {

public:

   double findMedianSortedArrays(vector<int>& nums1,vector<int>& nums2) {

       int m = nums1.size(), n = nums2.size();

       return (findKth(nums1, nums2, (m + n + 1) / 2) + findKth(nums1, nums2,(m + n + 2) / 2)) / 2.0;

    }

   int findKth(vector<int> nums1, vector<int> nums2, int k) {

       int m = nums1.size(), n = nums2.size();

       if (m > n) return findKth(nums2, nums1, k);

       if (m == 0) return nums2[k - 1];

       if (k == 1) return min(nums1[0], nums2[0]);

       int i = min(m, k / 2), j = min(n, k / 2);

       if (nums1[i - 1] > nums2[j - 1]) {

returnfindKth(nums1, vector<int>(nums2.begin() + j, nums2.end()), k - j);

// vector<int>(nums2.begin() + j,nums2.end())   就是数组从第j开始到结束！

       } else {

           return findKth(vector<int>(nums1.begin() + i, nums1.end()), nums2,k - i);

       }

       return 0;

    }

};