求两个有序数组中的中位数

时间要求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)
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值