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