Leetcode-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

二、思路分析

题目要求时间复杂度为log(m+n),则一般用分治法或二分查找法。

1、假设两序列总长度为n，如果n为奇数，则第n/2+1为中位数，如果n为偶数，则（n/2+n/2+1）/2是中位数。

2、序列从小到大排列，则第一个序列的前p个元素和第二个序列的前q个元素，我们有p+q = k-1，其中第一个序列的第p个元素和第二个序列的第q个元素都比总序列的第k个元素小，则第p+1或者第q+1则为我们要找的第k个数字。

3、利用二分法，假设p = k/2-1，q = k-p-1,如果第一序列的第p个元素小于第二序列的第q个元素，则我们舍弃第一序列的前p个元素，形成新的序列，即第一序列的起始点start1 = start1 + p，其中start1初始化为0，此时k = k-p，相反如果第二序列的第q个元素小于第一序列的第p个元素，则我们舍弃第二序列的前q个元素，将第二序列的起始点设为start2 = start2+q，k = k-q，如此递归下去，递归的终止条件为：较短序列的长度为0时，取较长序列的第k个元素，即为我们要找的，这里的较短序列和较长序列都是递归形成的，且k为递归形成的新值

三、代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    	int m = nums1.size();
    	int n = nums2.size();
    	int k = (m+n)/2;
    	if ((m+n)%2 == 0) {
    		return (findkth(nums1, m, nums2, n, 0, 0, k)+findkth(nums1, m, nums2, n, 0, 0, k+1))/2.0;
    	} else {
    		return findkth(nums1, m, nums2, n, 0, 0, k+1);
    	}
    }
private:
	double findkth(vector<int>& nums1, int m, vector<int>& nums2, int n, int start1, int start2, int k) {
		if (m > n) {
			return findkth(nums2, n, nums1, m, start2, start1, k);
		} 
		if (m == 0) {
			return nums2[start2 + k-1];
		}
		if (k == 1) {
			return min(nums1[start1], nums2[start2]);
		}
		int p = min(k/2, m);
		int q = k-p;
		if (nums1[start1+p-1] < nums2[start2+q-1]) {
			return findkth(nums1, m-p, nums2, n, start1+p, start2, k-p);
		} else {
			return findkth(nums1, m, nums2, n-q, start1, start2+q, k-q);
		}
	}
};