4. 寻找两个正序数组的中位数(二分查找)

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

代码:

func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
	length := len(nums1) + len(nums2)
	// k是第几个不是数组索引
	k := length/2 + 1

	if length%2 == 1 { // 奇数个
		return float64(findKnumber(nums1, nums2, k))
	} else { // 偶数个
		return float64(findKnumber(nums1, nums2, k)+findKnumber(nums1, nums2, k-1)) / 2
	}
	return 0
}

// 根据K值寻找第K小数
func findKnumber(m, n []int, k int) int {
	// 设置两数组的指针
	pointer1, pointer2 := 0, 0
	lm, ln := len(m), len(n)
	for {
		// 临界值,程序出口
		if pointer1 == len(m) { // m数组索引完成,取n数组中的值,k不是索引数所以-1
			return n[pointer2+k-1]
		}
		if pointer2 == len(n) { // n数组索引完成,取m数组中的值,k不是索引数所以-1
			return m[pointer1+k-1]
		}
		if k == 1 { // k不是索引数,为1时说明已经找到,取二者最小值
			return mix(m[pointer1], n[pointer2])
		}

		// k/2
		half := k / 2
		// 变更指针,指到k/2
		newPointer1, newPointer2 := mix(pointer1+half, lm)-1, mix(pointer2+half, ln)-1
		if m[newPointer1] <= n[newPointer2] {
			k = k - (newPointer1 - pointer1 + 1)
			pointer1 = newPointer1 + 1 // +1是因为newPointer对应的元素不为k小数
		} else {
			k = k - (newPointer2 - pointer2 + 1)
			pointer2 = newPointer2 + 1
		}
	}
}
func mix(x, y int) int {
	if x < y {
		return x
	}
	return y
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值