leetCode 寻找两个有序数组的中位数Golang解法

最新推荐文章于 2024-04-09 14:42:12 发布

gxsf

最新推荐文章于 2024-04-09 14:42:12 发布

阅读量421

点赞数

分类专栏： golang leetcode 随笔

本文链接：https://blog.csdn.net/bbdoudou/article/details/100101242

版权

golang 同时被 3 个专栏收录

5 篇文章 0 订阅

订阅专栏

leetcode

4 篇文章 0 订阅

订阅专栏

随笔

1 篇文章 0 订阅

订阅专栏

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

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

则中位数是 2.0
示例 2:

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

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

解：

func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
	aLen := len(nums1)
	bLen := len(nums2)
	allLen := aLen + bLen
	i, j := 0, 0
	var tmpNums []int
	if allLen%2 == 1 {
		for {
			if i < aLen && j < bLen {
				if nums1[i] <= nums2[j] {
					tmpNums = append(tmpNums, nums1[i])
					i++
				} else {
					tmpNums = append(tmpNums, nums2[j])
					j++
				}
			} else if i >= aLen && j < bLen{
				tmpNums = append(tmpNums, nums2[j])
				j++
			} else if j >= bLen && i < aLen {
				tmpNums = append(tmpNums, nums1[i])
				i++
			}
			if len(tmpNums) == allLen/2+1 {
				return float64(tmpNums[allLen/2+1-1])
			}
		}
	} else {
		for {
			if i < aLen && j < bLen {
				if nums1[i] <= nums2[j] {
					tmpNums = append(tmpNums, nums1[i])
					i++
				} else {
					tmpNums = append(tmpNums, nums2[j])
					j++
				}
			} else if i >= aLen && j < bLen {
				tmpNums = append(tmpNums, nums2[j])
				j++
			} else if j >= bLen && i < aLen {
				tmpNums = append(tmpNums, nums1[i])
				i++
			}
			if len(tmpNums) == allLen/2+1 {
                return float64((tmpNums[allLen/2+1-1] + tmpNums[allLen/2-1])) / 2.0
			}
		}
	}
	return 0
}

还可以优化一下的如果两个数组的数相等时，直接都append新数组里去，少一次循环