LeetCode 4. Median of Two Sorted Arrays

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/y12345678904/article/details/79946171

题目

这里写图片描述

思路

分治。在两个数组中寻找第k大的数。
如果nums1[k/2]<nums2[k/2],那么数字不会在nums1[:k/2+1]中,反之亦然。

代码

class Solution:
    def findKthNum(self, nums1, nums2, k):
        if not nums1: return nums2[k - 1]        

        if len(nums1) > len(nums2):
            return self.findKthNum(nums2, nums1, k)

        if k == 1:
            return min(nums1[0], nums2[0])   

        index1 = min(k // 2, len(nums1))
        index2 = k - index1

        if nums1[index1 - 1] <= nums2[index2 - 1]:
            return self.findKthNum(nums1[index1:], nums2, index2)
        else:
            return self.findKthNum(nums1, nums2[index2:], index1)

    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        k = len(nums1) + len(nums2)
        if k % 2:
            return self.findKthNum(nums1, nums2, k // 2 + 1)
        else:
            return (self.findKthNum(nums1, nums2, k // 2) + self.findKthNum(nums1, nums2, k // 2 + 1)) * 0.5

阅读更多

扫码向博主提问

sparksnail

非学,无以致疑;非问,无以广识
去开通我的Chat快问
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页