Go语言:LeetCode-寻找两个有序数组的中位数

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

分析

  1. 要找两个有序数组的中位数,首先要获得这两个数组合并后的有序数组。但是,我们并不需要得到长度为 (m+n) 的合并后的数组,因为我们只是需要找中位数,我们合并到 1/2 *(m+n)的长度即可。
  2. 利用指针快速有效的合并数组。对每个数组设置一个指针,刚开始都指向数组头,如果第一个数组的指针所指的数大于另一个数组的指针所指的数,那么该数组的指针向下移动一位。具体如以下图片所示。
    在这里插入图片描述由于蓝色指针所指的数比橙色指针所指的数要小,所以合并后的第一个数选择蓝色指针所指的数,下一步是蓝色指针往后移动一位,橙色指针位置保持不变。如下图所示。
    在这里插入图片描述
    蓝色指针所指的数仍然比橙色指针所指的数要小,所以合并后的第二个数仍然取蓝色指针所指的数,下一步依然是蓝色指针往后移动一位,橙色指针位置保持不变。如下图所示。
    在这里插入图片描述
    在上图中,蓝色指针所指的数比红色指针所指的数要大,所以合并后的第三个数是取橙色指针所指的数,下一步是蓝色指针保持不动,橙色指针往后移动一位,接着继续比较。
    在这里插入图片描述
    就是按照这种方式找到长度为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上通过的,因此在此分享一下。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值