Description
There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
You may assume nums1 and nums2 cannot be both empty.
Example 1:
1 | nums1 = [1, 3] |
Example 2:
1 | nums1 = [1, 2] |
思路
对于复杂度的要求加大了题目的难度,一般容易想到线性级别的解决方法,如果要降到对数级别,要用到二分查找法。这里的二分其实是针对切割的位置。参考了LeetCode上的解答。
为了简化问题,偶数和奇数个数的有序数组,可以综合成一种情况:数组长度为N时,index of L = (N-1)/2, and R is at N/2。 中位数的下标可以表示为(L + R)/2 = (A[(N-1)/2] + A[N/2])/2。函数中使A1是更长的数组。
设想数组中有更多的位置:
1 | A1: [# 1 # 2 # 3 # 4 # 5 #] (N1 = 5, N1_positions = 11) |
两个数组共有2N1 + 2N2 + 2个位置,每次分割后左右分别有N1 + N2个位置,最终的分割应保证L1 <= R1 && L1 <= R2 && L2 <= R1 && L2 <= R2,相当于L1 <= R2&&L2 <= R1。
在两种情况下需要对切割的位置做出调整:
L1>R2:说明A1左边大数过多,A1的切割点左移,同时A2切割点右移;
L2>R1:说明A2左边大数过多,A2的切割点左移,同时A1切割点右移;
否则,找到的即为中位数的位置,中位数等于(max(L1, L2) + min(R1, R2)) / 2。
C++代码
1 |
|
分析
复杂度:O(log(min(N1, N2)))
若下标超出数组界限,可以假设有两个额外的值 INT_MAX at A[-1] 和 INT_MAX at A[N],不影响结果并且简化了情况。