时间要求O(log(m+n))
package main
import "fmt"
func main() {
A:=[]int{1,2,3,4}
B:=[]int{5,5,6,7}
fmt.Println(findMid(A,B))
}
func findMid(nums1,nums2[]int)float32{
m,n:=len(nums1),len(nums2)
k:=(m+n+1)/2 //问题变成求第k小元素
//nums1=[1 2 3] nums2= [4 5] k=3
// if nums1[k/2-1] <nums2[k/2-1] //说明nums1[0]到nums1[k/2-1] 总共k/2 个数不在寻找范围之内,去掉k/2 个数,从剩余数组中查询第k/2 个元素
// 新数组变为 [2,3] 和 [4,5] k=3-1=2
// nums1[k/2-1]=2 nums2[k/2-1]=4 2<4 去掉k/2=1 个元素,k=1,
// [3] [4,5] if k==1 比较元素,返回最小的那一个 3
if (m+n)%2==1{
return float32(helper(nums1,nums2,k))
}else{
return (float32(helper(nums1,nums2,k))+float32(helper(nums1,nums2,k+1)))/2.0
}
}
func helper(nums1,nums2 []int,k int)int{
if len(nums1)==0{
return nums2[k-1]
}else if len(nums2)==0{
return nums1[k-1]
}
if k==1{
if nums1[0]<nums2[0]{
return nums1[0]
}else{
return nums2[0]
}
}
if nums1[k/2-1]<nums2[k/2-1]{
return helper(nums1[k/2:],nums2,k-k/2)
}else{
return helper(nums1,nums2[k/2:],k-k/2)
}
}