题意:
给出两个分别有序的数组,数组长度分别为m和n,设计一个求出这两个数组所有元素的中位数的算法,要求时间复杂度不大于O(log(m+n)).
解题思路:https://leetcode.com/problems/median-of-two-sorted-arrays/#/solutions
自己没有思路。这个方法非常好,作者讲的简单易懂,非常有条理。
代码:
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
double ans;
int m = nums1.size(), n = nums2.size();
vector<int> tmp; int t;
if(m > n){
tmp = nums1; t = m;
nums1 = nums2; m = n;
nums2 = tmp; n = t;
}
int l = 0, r = m, half = (m+n+1)>>1;
int max_of_left, min_of_right;
while(l <= r){
int i = (l+r) >> 1, j = half - i;
if(i < m && nums2[j-1] > nums1[i]) l = i+1;
else if(i > 0 && nums1[i-1] > nums2[j]) r = i-1;
else{
if(i == 0) max_of_left = nums2[j-1];
else if(j == 0) max_of_left = nums1[i-1];
else max_of_left = max(nums1[i-1], nums2[j-1]);
if((m+n)%2==1) ans = max_of_left;
if(i == m) min_of_right = nums2[j];
else if(j == n) min_of_right = nums1[i];
else min_of_right = min(nums1[i], nums2[j]);
if((m+n)%2==0) ans = (max_of_left + min_of_right) / 2.0;
break;
}
}
return ans;
}
};
看到问题不能想到其本质,不能很好地抽象出模型,要素之间的关系不能充分发现利用。