leetcode第4题 如何求出两个有序数组的中位数

leetcode原题大意，给定两个升序排列的有序数组，例如 nums1=[1,2], nums2=[3,4]那么，这两个有序数组的所有数字的中位数为 (2+3)/2 = 1.5，现在要求以O(log(m+n))的时间复杂度。

func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
	length := len(nums1) + len(nums2)
	if length%2 != 0 {
		return float64(helper(nums1, nums2, (length+1)/2))
	} else {
		x := float64(helper(nums1, nums2, length/2))
		y := float64(helper(nums1, nums2, length/2+1))
		return (x + y) / 2
	}
}
func helper(nums1 []int, nums2 []int, k int) int {
	if len(nums1) == 0 {
		return nums2[k-1]
	}
	if len(nums2) == 0 {
		return nums1[k-1]
	}
	if k == 1 {
		return min(nums1[0], nums2[0])
	}
	countOne := min(k/2, len(nums1))
	maxOne := nums1[countOne-1]
	countTwo := min(k/2, len(nums2))
	maxTwo := nums2[countTwo-1]
	if maxOne < maxTwo {
		return helper(nums1[countOne:], nums2, k-countOne)
	} else {
		return helper(nums1, nums2[countTwo:], k-countTwo)
	}
}

func min(a, b int) int {
	if a < b {
		return a
	}
	return b
}