1.题目介绍:
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log(m+n)) 。
示例1
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
来源:力扣(LeetCode)
2.题目分析:
2.1 题目转化
- 题目本身其实并不难理解,就是一个常见的合并然后查找中位数的问题,但是题目限定算法时间复杂度为O(log(m+n)) ,这就限制了我们不能使用最容易想到的直接按序归并到中间数,因为这样解决的时间复杂度时O(m+n).
- 对于基本的查找中位数的的问题,我们首先要考虑的是数据个数是奇数还是偶数,并根据数据个数决定中位数是直接取数组中中间值,还是取两个中间值的平均值。回过头来看题目,查找中位数,其实就是查找第k个大小的数的问题,然后题目所给的数组本身就是有序的,然后联想到我们要求的复杂度由log,很容易可以想到是采用二分查找。但是怎么使用二分查找呢?
- 我们可以首先简化题目为在两个长度分别为
length1
和length2
的数组中查找第k小的数据。
那么我们此时可以按照以下步骤进行处理:
- 我们分别标记出两个数组中每个数组第
k/2
大小的值,并将其进行比较,可以知道较小的值得左边的数必然不会存在总体数组中第k个小的数 - 然后我们可以将选定的数组的前
k/2
个数在下次查找中排除,并对k
进行更新,因为此时已经排除了k/2
个数,所以此时在剩下的数组中查找第k-k/2
个小的数 - 然后重复1~2步骤,直到
k=1
时此时就可以确定值为较小的一个标记值
- 基本思想步骤确定之后我们需要确定一些边界条件,因为二分查找是一个循环的过程,首先就是如果其中一个数组已经排除完,那么中位数肯定在另外一个数组中,此时我们可以直接跳出循环,其次就是当k等于1时,此时找的是最小的一个数,因此我们只需要返回较小的标记点的值就可以了
进行上面的思考后进行解决问题就完成了吗,并没有,在实际处理中因为每次计算标记值都是需要当前标记值向前移动k/2
个位置,因为在解题中对数组的访问采用的是下标访问,那么我们肯定要考虑下标溢出的问题,如下面就是未考虑下标溢出报错,原因是在第一轮排除后,index1=0,index2=0
,在第二次进行标记值计算时会产生index1+=1,此时数组下标越界报错。所以在实际编码中必须对index1
和index2
的赋值加以严格控制,以index1来说,他的赋值是这样的
int newIndex1 = min(index1 + k / 2 - 1, m - 1);
另外,在解题中,我对于k的更新起初是k-=k/2;
这其实是不合理的。因为在上面的newIndex
更新中决定了下次标记值是哪一个,而k的更新应该是取决于标记值的。所以k的更新如最终代码所示。
2.2 实例分析
用下面这个例子来说明算法就是,首先有AB两