2020年8月6日 周四 天气晴 【不悲叹过去,不荒废现在,不惧怕未来】
1. 题目简介
这道题还是有一定难度的,但是也非常的经典,从LeetCode排名第4就能看出来。题目要求算法的时间复杂度为 O(log(m + n)),这里介绍两种方法,都是基于二分法,一种复杂度是 O(log(m + n)),另外一种复杂度仅为 O(log(min(m, n))。
2. 第k小数法
这道题求中位数,可以看作是求第k小数的特例,因为数组的中位数就等于第(m + n + 1)/ 2 小的数和第(m + n + 2)/ 2 小的数之和的平均值(如果是奇数,这两个数就是同一个数)。那么如何求两个数组的第k小数呢?就是利用分治的思想,借助二分法来实现。具体可以看这里的解法三,我觉得大佬讲的很清晰,自己就不去赘述了。
时间复杂度 O(log(m + n))
2.1 递归法
递归法相对好理解一点,并且由于这个递归属于尾递归,所以编译器不需要不停地堆栈,因此空间复杂度为 O(1)。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size(), n = nums2.size();
int left = (m + n + 1) / 2;
int right = (m + n + 2) / 2;
// 将偶数和奇数的情况合并,如果是奇数,会求两次同样的 k
return ((double)findK(nums1, 0, m - 1, nums2, 0, n - 1, left) + findK(nums1, 0, m - 1, nums2, 0, n - 1, right)) / 2.0;
}
int findK(vector<int>& nums1, int start1, int end1, vector<int>& nums2, int start2, int end2, int k) {
int len1 = end1 - start1 + 1, len2 = end2 - start2