每日编程中遇到任何疑问、意见、建议请公众号留言或直接撩Q474356284(备注每日编程)
今日问题:
给定两个大小为 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
解决方法:
以下算法解释和思路参考亮哥博客文章(已授权):
http://windliang.cc/2018/07/18/leetCode-4-Median-of-Two-Sorted-Arrays/
我们首先理一下中位数的定义是什么
中位数(又称中值,英语:Median),统计学中的专有名词,代表一个样本、种群或概率分布中的一个数值,其可将数值集合划分为相等的上下两部分。
所以我们只需要将数组进行切。
一个长度为 m 的数组,有 0 到 m 总共 m + 1 个位置可以切。
我们把数组 A 和数组 B 分别在 i 和 j 进行切割。
将 i 的左边和 j 的左边组合成「左半部分」,将 i 的右边和 j 的右边组合成「右半部分」。
当 A 数组和 B 数组的总长度是偶数时,如果我们能够保证
左半部分的长度等于右半部分
i + j = m - i + n - j , 也就是 j = ( m + n ) / 2 – i
b)左半部分最大的值小于等于右半部分最小的值
max (A [ i - 1 ] , B [ j - 1 ])<=min ( A [ i ] , B [ j ]))
那么,中位数就可以表示如下(左半部分最大值 + 右半部分最大值 )/2 。
(max ( A [ i - 1 ] , B [ j - 1 ])+min ( A [ i ] , B [ j ])) / 2
当 A 数组和 B 数组的总长度是奇数时,如果我们能够保证
a)左半部分的长度比右半部分大 1
i + j = m - i + n - j + 1也就是j = ( m + n + 1) / 2 – i
b)左半部分最大的值小于等于右半部分最小的值
max ( A [ i - 1 ] , B [ j - 1 ]))<= min ( A [ i ] , B [ j ]))
那么,中位数就是
左半部分最大值,也就是左半部比右半部分多出的那一个数。
max ( A [ i - 1 ] , B [ j - 1 ])
上边的第一个条件我们其实可以合并为 j = ( m + n + 1) / 2 - i,
因为如果 m + n 是偶数,由于我们取的是 int 值,所以加 1 也不会影响结果。当然,由于 0 <= i <= m ,为了保证 0 <= j <= n ,我们必须保证 m <= n 。
m≤n,i(m+m+1)/2−m= 0
m≤n,i>0, j = (m+n+1)/2−i ≤ (n+n+1)/2−i < (n+n+1)/2= n
最后一步由于是 int 间的运算,所以 1 / 2 = 0。
而对于第二个条件,奇数和偶数的情况是一样的,我们进一步分析。为了保证
max ( A [ i - 1 ] , B [ j - 1 ])) <= min ( A [ i ] , B [ j ])),因为 A 数组和 B 数组是有序的,所以 A[ i - 1 ] <= A [ i ],B [ i - 1 ] <= B [ i ] 这是天然的,所以我们只需要保证 B [ j - 1 ] < = A [ i ] 和 A [ i - 1 ]<= B [ j ] 所以我们分两种情况讨论:
B [ j - 1 ] > A [ i ],并且为了不越界,要保证 j != 0,i != m
此时很明显,我们需要增加 i ,为了数量的平衡还要减少 j ,
幸运的是 j = ( m + n +1) / 2 - i,i 增大,j 自然会减少。
A [ i - 1 ] > B [ j ] ,并且为了不越界,要保证 i != 0,j != n
此时和上边的情况相反,我们要减少 i ,增大 j 。
上边两种情况,我们把边界都排除了,需要单独讨论。
当 i = 0 , 或者 j = 0 ,也就是切在了最前边。
此时左半部分当 j = 0 时,最大的值就是 A [ i - 1] ;当 i = 0 时 最大的值就是 B [ j - 1] 。右半部分最小值和之前一样。
当 i = m 或者 j = n ,也就是切在了最后边。
此时左半部分最大值和之前一样。右半部分当 j = n 时,最小值就是 A [ i ] ;
当 i = m 时,最小值就是B [ j ] 。
所有的思路都理清了,最后一个问题,增加i 的方式。当然用二分了。初始化 i 为中间的值,然后减半找中间的,减半找中间的,减半找中间的直到答案。
时间复杂度:我们对较短的数组进行了二分查找,所以时间复杂度是
O(log(min(m,n)))。
空间复杂度:只有一些固定的变量,和数组长度无关,所以空间复杂度是 O ( 1 ) 。
C++代码:
C代码:
Java代码:
明日题目预告:
最长回文子串
给定一个字符串 s,找到 s 中最长的回文子串。你可以假设 s 的最大长度为1000。
示例 1:
输入:"babad"
输出:"bab"
注意:"aba"也是一个有效答案。
示例 2:
输入:"cbbd"
输出:"bb"