Go语言的LeetCode刷题之旅-004-两个排序数组的中位数

最新推荐文章于 2020-11-27 20:34:12 发布

原创最新推荐文章于 2020-11-27 20:34:12 发布 · 847 阅读

1 ·

CC 4.0 BY-SA版权

LeetCode 同时被 3 个专栏收录

7 篇文章

订阅专栏

算法

7 篇文章

订阅专栏

Golang

5 篇文章

订阅专栏

本文介绍了一种使用Go语言解决LeetCode上经典问题——寻找两个已排序数组中位数的方法。该算法需达到O(log(m+n))的时间复杂度。文章提供了两种解法，一种通过合并两个数组后求中位数，另一种则直接在原数组基础上进行排序后求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Go语言的LeetCode刷题之旅-004-两个排序数组的中位数

题目：

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。

请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。

示例 1:

nums1 = [1, 3]
nums2 = [2]

中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

中位数是 (2 + 3)/2 = 2.5

思路：

输入两个已经排序好的整数切片，合并两个切片，返回合并后切片的中位数。

所以，题目的关键是，如何高效地合并切片。

Go代码：

package problem0004

func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
    nums := combine(nums1, nums2)
    return medianOf(nums)
}

func combine(mis, njs []int) []int {
    lenMis, i := len(mis), 0
    lenNjs, j := len(njs), 0
    res := make([]int, lenMis+lenNjs)

    for k := 0; k < lenMis+lenNjs; k++ {
        if i == lenMis ||
            (i < lenMis && j < lenNjs && mis[i] > njs[j]) {
            res[k] = njs[j]
            j++
            continue
        }

        if j == lenNjs ||
            (i < lenMis && j < lenNjs && mis[i] <= njs[j]) {
            res[k] = mis[i]
            i++
        }
    }

    return res
}

func medianOf(nums []int) float64 {
    l := len(nums)

    if l == 0 {
        panic("切片的长度为0，无法求解中位数。")
    }

    if l%2 == 0 {
        return float64(nums[l/2]+nums[l/2-1]) / 2.0
    }

    return float64(nums[l/2])
}

解法2：

func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
        for _,k:=range nums2{
            nums1=append(nums1,k)
        }
    x :=len(nums1)
    for i:=0;i<x-1;i++{
        for j:=0;j<len(nums1)-1-i;j++{
            if nums1[j]>nums1[j+1]{
                nums1[j],nums1[j+1]=nums1[j+1],nums1[j]
            }
        }
    }
    var res float64
    if(x%2==0){
        y :=x/2
        res =  float64((nums1[y]+nums1[y-1]))/2
    }else {
        y:=x/2
        res = float64(nums1[y])
    }
    return res
}