LeetCode 第四题：寻找两个正序数组的中位数 【4/1000 】【python + go】

题目描述：

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

示例：

输入：nums1 = [1,3], nums2 = [2]

输出：2.00000

解释：合并数组 = [1,2,3] ，中位数 2

提示：

nums1.length == m

nums2.length == n

0 <= m <= 1000

0 <= n <= 1000

1 <= m + n <= 2000

-106 <= nums1[i], nums2[i] <= 106

进阶：

你能设计一个时间复杂度为 O(log(m + n)) 的算法解决此问题吗？

Python 解法：

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        nums = nums1 + nums2
        nums.sort()
        n = len(nums)
        if n % 2 == 1:
            return nums[n // 2]
        else:
            return (nums[n // 2 - 1] + nums[n // 2]) / 2.0

Go 解法：

package main
 
import (
    "sort"
)
 
func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
    nums := append(nums1, nums2...)
    sort.Ints(nums)
    n := len(nums)
    if n%2 == 1 {
        return float64(nums[n/2])
    } else {
        return float64(nums[n/2-1]+nums[n/2]) / 2.0
    }
}
 
func main() {
    // 测试用例
}

这两个解法都是将两个数组合并，然后排序，最后根据排序后数组的长度是奇数还是偶数来计算中位数。如果是奇数，中位数就是中间的元素；如果是偶数，中位数是中间两个元素的平均值。