题目链接
暴力
n
l
o
g
(
n
)
nlog(n)
nlog(n)
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
vector<double> C;
for(int i = 0; i < nums1.size(); i++) C.push_back(nums1[i]);
for(int i = 0; i < nums2.size(); i++) C.push_back(nums2[i]);
sort(C.begin(),C.end());
if(C.size()&1) return C[C.size()>>1];
else return (C[C.size()>>1] + C[(C.size()>>1)-1])/2.0;
}
};
二分 O ( l o g m i n ( n , m ) ) O(logmin(n,m)) O(logmin(n,m))
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);
int L = 0, R = N2<<1;
while (L <= R) {
int mid2 = (L + R) >> 1;
int mid1 = N1 + N2 - mid2;
double L1 = (mid1 == 0) ? INT_MIN : nums1[(mid1-1)/2];
double L2 = (mid2 == 0) ? INT_MIN : nums2[(mid2-1)/2];
double R1 = (mid1 == N1 * 2) ? INT_MAX : nums1[(mid1)/2];
double R2 = (mid2 == N2 * 2) ? INT_MAX : nums2[(mid2)/2];
if (L1 > R2) L = mid2 + 1;
else if (L2 > R1) R = mid2 - 1;
else return (max(L1,L2) + min(R1, R2)) / 2;
}
return 0;
}
};