LeetCode 4. Median of Two Sorted Arrays

sparksnail

于 2018-04-15 01:35:57 发布

阅读量148

点赞数

CC 4.0 BY-SA版权

分类专栏： OJ-LeetCode 算法-数组

本文链接：https://blog.youkuaiyun.com/y12345678904/article/details/79946171

OJ-LeetCode 同时被 2 个专栏收录

192 篇文章

订阅专栏

算法-数组

11 篇文章

订阅专栏

该博客讨论了LeetCode第4题的解决方案，通过分治策略在两个已排序数组中找到中位数。主要思路是在两个数组中分别查找第k大的数，根据比较结果排除部分元素，逐步逼近中位数。

题目

这里写图片描述

思路

分治。在两个数组中寻找第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