二分法的hard题目。
LeetCode4. 寻找两个有序数组的中位数
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
难就难在时间复杂度上。如果不考虑这点,用归并排序的merge思路解决,复杂度O(m+n).
# 归并的merge步骤,O(m+n)
class Solution:
def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
m, n = len(nums1), len(nums2)
#奇数时候,mid是中间值索引,偶数是后一个,所以需要pre
mid = (m+n)//2
# 添加哨兵,减少判断次数
nums1.append(float('inf'))
nums2.append(float('inf'))
i, j = 0, 0
pre = res = 0
for k in range(mid+1):
pre = res
if nums1[i] <= nums2[j]:
res = nums1[i]
i += 1
else:
res = nums2[j]
j += 1
#恢复原样
nums1.pop()
nums2.pop()
return res if (m+n) % 2 == 1 else (res+pre)/2
时间复杂度有log基本就是二分了。
leetcode上有多种解法,每次去掉K/2数字(K是中值位置)和将两个数组转换成一个数组二分。
每次去掉K/2边界条件令人头疼,知道流程我也写半天(也可能是我太菜了,写不出好看的表达式),可以参考这里。
这里记录转换的思路,非常巧妙。参考LeetCode Median of Two Sorted Arrays 在两个已排列的数组中找出中位数。时间复杂度为O(log(min(N,M))。
已知K为中值的位置,那么假设在A数组取前a个, 在B数组中取的个数就被相应确定为K-a;所以问题被转换为,查找A数组中a位置。
A中有m个数字,索引为0,m-1。将当前索引划分给右侧,可选的划分位置为0…m。
如果左边序列的最大值小于等于右边序列的最小值那么这就是我们要找的划分。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
//取小的数组进行二分搜索。
if(nums1.size() > nums2.size())
return findMedianSortedArrays(nums2, nums1);
int m = nums1.size();
int n = nums2.size();
if(m==0) return (nums2[n/2]+nums2[(n-1)/2])/2.0;
int mid = (m+n)/2; //比真正的中间值小1
int left = 0, right=m;
int cut1, cut2, L1, L2, R1, R2;
while(true){
//cut1-1,cut1。左边共cut1个数字。cut1+cut2=mid个左值
cut1 = left + (right-left)/2;
cut2 = mid-cut1;
//用最大最小,避免处理边界
L1 = cut1==0? INT_MIN:nums1[cut1-1];
L2 = cut2==0? INT_MIN:nums2[cut2-1];
R1 = cut1==m? INT_MAX:nums1[cut1];
R2 = cut2==n? INT_MAX:nums2[cut2];
//数组1的左值大于数组2的右值,说明划分区间太大
if(L1>R2) right = cut1-1;
else if(L2>R1) left = cut1+1;
else{
//如果是奇数个, mid比真正的中值少一个,所以在右边取
if((m+n)%2==1) return min(R1, R2);
else return ((max(L1, L2)+min(R1, R2))/2.0);
}
}
}
};