LeetCode0004 寻找两个有序数组的中位数

最新推荐文章于 2021-10-12 00:00:00 发布

VIP文章「已注销」

最新推荐文章于 2021-10-12 00:00:00 发布

阅读量521

点赞数 1

分类专栏： LeetCode题库——Python3解答文章标签：分治二分法 Python3

本文链接：https://blog.csdn.net/LoveHYZH/article/details/92760709

版权

题目描述

给定两个大小为 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

求解思路

【注】思路主要是给本人自己看的，以便忘了能记起。

首先，我们需要理解 “中位数的作用是什么”。在统计中，中位数被用来：

将一个集合划分为两个长度相等的子集，其中一个子集中的元素总是大于另一个子集中的元素。

        如果是两个等长的数组，那么我们可以分别求2个数组的中位数，比较大小后，各自删除一半，反复如此，直到两边数组都只剩一个元素为止，最终中位数就是这两个元素的平均数。
        然而，此题没要要求两个数组长度必须一致，难就难在这里，所以只能模拟枚举。整体的中位数把元素分割成 $\bf \blue {left}$ 和 $\bf \blue{right}$ 两个集合，我们就要探测在这两个集合中nums1和nums2数组各有多少个元素在里面。
        简单的方法是类似合并两个有序列表的算法，时间复杂度是 $O (m i n (m, n))$ ，但是题目要求的是 $O (l o g (m + n))$ ，显然是要用到递归，不然时间复杂度不可能出现对数。
        怎么用递归呢，参考两个数组长度相等找中位数的算法，每次删去一部分，直到最后剩下结果。现在数组长度不相等了，这种二分元素显然行不通，所以退而求其次，二分索引，并且是长度较小的数组的索引。为什么是较短的数组？在后面编码时你会发现，探测长数组的索引可能会导致一些麻烦，需要额外的工作。
        那么，二分所引怎么用呢？通常情况，是设置两个哨兵 $i\_max$ 和 $i\_min$ ，探测位置 $i=func(i\_min,i\_max)$ ，通过判断 $i$ 是偏大还是偏小，来调整 $i\_max$ 或 $i\_min$ 值，然后继续这一步，直到找到满足中位数要求的 $i$ 。

-------------------------------------下面开始具体算法-------------------------------------

首先，根据数组长度进行排序，令短的为nums1,长的为nums2。m = len(nums1), n = len(nums2)
我们将nums1分成两个部分，左边 $i$ 个元素，右边 $m - i$ 个元素，其中 $0\le i \le m$ ：