給定兩個大小為 m 和 n 的有序數組?nums1
?和?nums2
。
請你找出這兩個有序數組的中位數,并且要求算法的時間復雜度為?O(log(m + n))。
你可以假設?nums1
?和?nums2
?不會同時為空。
示例 1:
nums1 = [1, 3] nums2 = [2]則中位數是 2.0
示例 2:
nums1 = [1, 2] nums2 = [3, 4]則中位數是 (2 + 3)/2 = 2.5
class Solution {public double findMedianSortedArrays(int[] A, int[] B) {int m = A.length;int n = B.length;if (m > n) { // to ensure m<=nint[] temp = A; A = B; B = temp;int tmp = m; m = n; n = tmp;}int iMin = 0, iMax = m, halfLen = (m + n + 1) / 2;while (iMin <= iMax) {int i = (iMin + iMax) / 2;int j = halfLen - i;if (i < iMax && B[j-1] > A[i]){iMin = i + 1; // i is too small }else if (i > iMin && A[i-1] > B[j]) {iMax = i - 1; // i is too big }else { // i is perfectint maxLeft = 0;if (i == 0) { maxLeft = B[j-1]; }else if (j == 0) { maxLeft = A[i-1]; }else { maxLeft = Math.max(A[i-1], B[j-1]); }if ( (m + n) % 2 == 1 ) { return maxLeft; }int minRight = 0;if (i == m) { minRight = B[j]; }else if (j == n) { minRight = A[i]; }else { minRight = Math.min(B[j], A[i]); }return (maxLeft + minRight) / 2.0;}}return 0.0;} }
?