题目:给定两个大小为 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
分析:
- 要找两个有序数组的中位数,首先要获得这两个数组合并后的有序数组。但是,我们并不需要得到长度为 (m+n) 的合并后的数组,因为我们只是需要找中位数,我们合并到 1/2 *(m+n)的长度即可。
- 利用指针快速有效的合并数组。对每个数组设置一个指针,刚开始都指向数组头,如果第一个数组的指针所指的数大于另一个数组的指针所指的数,那么该数组的指针向下移动一位。具体如以下图片所示。
由于蓝色指针所指的数比橙色指针所指的数要小,所以合并后的第一个数选择蓝色指针所指的数,下一步是蓝色指针往后移动一位,橙色指针位置保持不变。如下图所示。
蓝色指针所指的数仍然比橙色指针所指的数要小,所以合并后的第二个数仍然取蓝色指针所指的数,下一步依然是蓝色指针往后移动一位,橙色指针位置保持不变。如下图所示。
在上图中,蓝色指针所指的数比红色指针所指的数要大,所以合并后的第三个数是取橙色指针所指的数,下一步是蓝色指针保持不动,橙色指针往后移动一位,接着继续比较。
就是按照这种方式找到长度为1/2*(m+n)的合并后数组即可。具体实现如下代码所示:
func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
var nums []int
len1 := len(nums1)
len2 := len(nums2)
lenSum := len1 + len2 //获得两个数组的长度之和
var i, j, k int
var res float64
var tag int
for {
//循环一直到有一个数组已经到尾了还是没有找到中位数就退出循环
if i != len1 && j != len2 {
if nums1[i] < nums2[j] {
nums = append(nums, nums1[i])
i++
} else {
nums = append(nums, nums2[j])
j++
}
k = len(nums)
//判断合并后数组的长度是否为两数组之和的一半
if k == lenSum/2+1 {
if lenSum%2 == 0 {
res = (float64)((nums[k-2] + nums[k-1])) / 2.0
} else {
res = (float64)(nums[k-1])
}
break
}
} else if i == len1 { //如果第一个数组到尾了
tag = 1
break
} else if j == len2 { //如果第二个数组到尾了
tag = 2
break
}
}
//对特殊情况进行处理
if tag == 1 {
for {
nums = append(nums, nums2[j])
j++
k = len(nums)
if k == lenSum/2+1 {
if lenSum%2 == 0 {
res = (float64)((nums[k-2] + nums[k-1])) / 2.0
} else {
res = (float64)(nums[k-1])
}
break
}
}
} else if tag == 2 {
for {
nums = append(nums, nums1[i])
i++
k = len(nums)
if k == lenSum/2+1 {
if lenSum%2 == 0 {
res = (float64)((nums[k-2] + nums[k-1])) / 2.0
} else {
res = (float64)(nums[k-1])
}
break
}
}
}
return res
}
这种算法的时间复杂度是 O(m+n) , 没有达到题目要求的 O(log(m+n)) 。但是这种方法也是能够在LeetCode上通过的,因此在此分享一下。