寻找两个有效数组的的中位数:link
1.题目分析
1.题目中有一个提示,要求时间复杂度是log的应该会想到是二分
2.有两个数组确定了一个数组划分的位置就可以确定另一个数组划分的位置。
很棒的题解link
1.划分的位置要满足条件
L1 < R1 && L2 < R2 && L1 < R2 && L2 < R1
;前两个条件天然满足
2.为了让点1中前两个条件天然满足,在处理边界条件时可以将
L1, L2 = INT_MIN, R1, R2 = INT_MAX
;
3.对于这种会切到数和间隔的情况,采用虚拟位置的方法。
2.示例代码
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int N1 = nums1.size();
int N2 = nums2.size();
if(N1 > N2)
return findMedianSortedArrays(nums2, nums1);//make sure the size of nums1 is smaller
int left = 0, right = 2 * N1;
int L1, R1, L2, R2;
while(1){
int mid1 = (left + right) / 2;
int mid2 = N1 + N2 - mid1;
//calculate L1, R1, L2, R2
L1 = (mid1 == 0) ? INT_MIN : nums1[(mid1 - 1) / 2];
R1 = (mid1 == 2 * N1) ? INT_MAX : nums1[mid1 / 2];
L2 = (mid2 == 0) ? INT_MIN : nums2[(mid2 - 1) / 2];
R2 = (mid2 == 2 * N2) ? INT_MAX : nums2[mid2 / 2];
if(L1 > R2)
right = mid1 - 1;
else if(L2 > R1)
left = mid1 + 1;
else
return (max(L1, L2) + min(R1, R2)) / 2.0;
}
return -1;//never happen
}
};