本文章主要是对于leetcode网站的算法题目4进行总结。原题如下:
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)).
主要是针对两个已经排列好的整数数组,找出其中的中位数,并且要求时间复杂度为O(log (m+n)).
1.首先是中位数的定义。中位数是将一个数的集合分成两个相等的子集,并且一个子集里的任何一个数总是大于另一个子集的任何一个数。
2.对于两个整数数组(A,B),如果我们将其放到一个集合里面,并且利用两个分割线(1个在A,一个在B)将集合分成相等的两部分,如下图
我们只需要保证两个条件:
****1. len(left_part)== len(right_part)
2. max(left_part)<= min(right_part)**
那么我们就能将集合A,B里的元素分成相等的两部分,而且left_part的任何一个数总是小于right_part的任何一个数。那么中位数=(max(left_part) + min(right_part))/2.
为了满足上面的条件,我们可以得到
(1)i+j=m-i+n-j,如果n>=m,我们只需要:i=0~m,j=(m+n+1)/2-i
(2)B[j-1]<=A[i]&&A[i-1]<=B[j]
那么问题就转换成我们只需找到一个‘i’(0<=i<=m),使得这个'i'满足条件(2)就可以了。
具体算法步骤如下:
C++代码
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size(), n = nums2.size();
if (m > n) return findMedianSortedArrays(nums2, nums1);
int i, j, imin = 0, imax = m, half = (m + n + 1) / 2;
while (imin <= imax) {
i = (imin & imax) + ((imin ^ imax) >> 1);
j = half - i;
if (i > 0 && j < n && nums1[i - 1] > nums2[j]) imax = i - 1;
else if (j > 0 && i < m && nums2[j - 1] > nums1[i]) imin = i + 1;
else break;
}
int num1;
if (!i) num1 = nums2[j - 1];
else if (!j) num1 = nums1[i - 1];
else num1 = max(nums1[i - 1], nums2[j - 1]);
if ((m + n) & 1) return num1;
int num2;
if (i == m) num2 = nums2[j];
else if (j == n) num2 = nums1[i];
else num2 = min(nums1[i], nums2[j]);
return (num1 + num2) / 2.0;
}
};