這個題好難,主要是有時間復(fù)雜度 O(log(m+n)) 的限制。
思路:
把問題轉(zhuǎn)化成求兩個數(shù)組中求第K小的數(shù)即可。
用分治法恃逻,若A[k/2-1] >B[k/2-1] (即A的第k/2個數(shù)大于B的第k/2個數(shù))
雏搂,則第K小的數(shù)不在B的前k/2個數(shù)中,同理B>A寇损。
由此凸郑,每進行比較一次,都會有k/2個數(shù)被篩選下去矛市,一直到最后就可以找到了芙沥。
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int k = nums1.size() + nums2.size();
if(k%2==0) return ( findKth(nums1,0,nums2,0,k/2) + findKth(nums1,0,nums2,0,k/2+1)) / 2.0;
else return findKth(nums1,0,nums2,0,k/2+1);
}
int findKth(vector<int>& nums1,int i,vector<int>& nums2,int j,int k){
if(i >= nums1.size()) return nums2[j + k-1];
if(j >= nums2.size()) return nums1[i + k-1];
if(k == 1) return min(nums1[i],nums2[j]);
int a = i+k/2-1 >= nums1.size() ? INT_MAX :nums1[i+k/2-1];
int b = j+k/2-1 >= nums2.size() ? INT_MAX:nums2[j+k/2-1];
if(a >= b) return findKth(nums1,i,nums2,j+k/2,k-k/2);
else return findKth(nums1,i+k/2,nums2,j,k-k/2);
}
注:開始智障了,沒有用INT_MAX浊吏,用的是10000而昨,結(jié)果沒通過,還找了很久的bug.....
太慢了找田,還可以進行優(yōu)化的(數(shù)組的K/2位置的數(shù)字要不要去除)歌憨。等研究完最快的代碼再來填坑。