继续刷LeetCode 热题 HOT 100 的题目,并且在博客更新我的solutions。在csdn博客中我会尽量用文字解释清楚,相关Java代码大家可以前往我的个人博客jinhuaiyu.com中查看。
今天这道题本来以为很容易,但是还是搞了很久。可能是coding的不够多,数据结构和算法忘了很多,希望后面能慢慢变快吧。
时间:2022.1.7
题目:寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
示例 4:
输入:nums1 = [], nums2 = [1]
输出:1.00000
示例 5:
输入:nums1 = [2], nums2 = []
输出:2.00000
提示:
nums1.length m
nums2.length n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
solution 1:
一开始只想到O(m+n)复杂度的方法,即通过一轮遍历将两个数组中任意一个删到只剩下一个或删空。用首位指针模拟删除数组(不是真的删除),每次循环中进行两轮比较:比较两个数组的首指针,小的右移(模拟删除);比较两个数组的尾指针,大的左移。于是每一轮循环都可以删除一个最小和一个最大的数,保持当前中位数仍然是原来的中位数。
循环结束后分成三种情况(每种情况再分奇偶讨论):
1、一个数组遍历完了,另一个数组还剩下,此时中位数结束剩下数组的中位数;
2、两个数组都只剩一个数字,中位数就是平均值;
3、一个数组遍历到只剩一个数字,另一个还剩下不止一个。较多的数组如果为奇数个,则取中间三个以及只剩一个的数组中的那个数,对这四个数取中位数。较多的数组如果为偶数个,则取中间两个以及只剩一个的数组中的那个数,对这三个数取中位数。
最后,测试没通过……因为可能输入的两个数组中有一个为空,于是在最开始打了个补丁判断特殊情况。这样就可以通过了,只不过离题目要求的复杂度还有一定距离。
solution 2:
看到O(logn)这种复杂度的,会往二分查找上思考。于是我视图从solution 1中插入二分查找来优化。
第一次尝试:
设置一个待删除数量,也就是说我们要删中位数前面这么多的数,中位数自然就出现了(这里我们不再两边一起删了,删一边就行了)。设置指针指向头部模拟删除。于是我试图每次循环比较两个数组二分处的数的大小,将小的那个数组的前一半删除,每次删除前检查待删除数量是不是用完了并更新它。但这种方法删到最后不够删了就会有bug,还要特殊处理,但是这块我越写越麻烦……最后abandon。
第二次尝试:
那么问题出在哪里呢?二分查找难道分的不是数组?我承认我最后还是看了一些其他人的解答……原来的二分查找分的是第一次尝试中的待删除数量。在第一次尝试中,我们无法把控每次删多少,所以最后首尾的时候非常复杂,既然这样,我们完全可以控制每次就删二分之一当前的待删除数量,直到它被删到只剩1个或者某个数组为空。
换句话说,我们要找中位数,其实就是找第k小的数(或者第k和第k+1两个数),每次删除前k/2小的数,然后更新k,可以通过递归得到剩下的数组离第k-k/2小的数。
怎么删除k/2小的数呢?可以比较两个数组中第k/2位置的数字,小的那个的前面k/2被删除。如果有数组k/2越界,那就只能删另外一个数组(两个数组不可能同时越界,数组大小和k是有关的)。
至此,一切都已经很明朗了。我们回到最开始,如何区分奇偶,前面一直没解决这个问题,事实上,用于int的除法的特殊性质,我们可以不用分类讨论这两种情况。找到下标(m + n - 1)/ 2和下标(m + n )/ 2元素,然后求两数的平均值即是中位数(奇数情况下这两个指向同一个位置)。
Finally,带有详细注释的代码放在我的个人博客http://jinhuaiyu.com/leetcode-median-of-two-sorted-arrays/
LeetCode 热题 HOT 100 第4天: “寻找两个正序数组的中位数”
最新推荐文章于 2024-08-21 10:11:50 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)