寻找两个正序数组的中位数

给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2

输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5

  • 思路1:合并数组,利用两个数组是有序合并数组,然后获取中位数
private fun getMiddleNum(nums1: IntArray, nums2: IntArray): Double {
        val resultArray = mutableListOf<Int>()
        var n1 = 0
        var n2 = 0
        // 合并数组
        while (n1 < nums1.size && n2 < nums2.size) {
            when {
                nums1[n1] < nums2[n2] -> {
                    resultArray.add(nums1[n1++])
                }
                nums1[n1] > nums2[n2] -> {
                    resultArray.add(nums2[n2++])
                }
                else -> {
                    resultArray.add(nums1[n1++])
                    resultArray.add(nums2[n2++])
                }
            }
        }
        while (n1 < nums1.size) {
            resultArray.add(nums1[n1++])
        }
        while (n2 < nums2.size) {
            resultArray.add(nums2[n2++])
        }
        // 返回中位数
        return when {
            resultArray.size == 0 -> {
                0.0
            }
            resultArray.size % 2 == 0 -> {
                val i = resultArray.size / 2
                (resultArray[i] + resultArray[i - 1]) / 2.toDouble()
            }
            else -> {
                val i = resultArray.size / 2
                resultArray[i].toDouble()
            }
        }
    }
  • 思路2:知道两个集合的总长度,根据角标获取应该获取对应数组的位置!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值