LeetCode 4. Median of Two Sorted Arrays

Descripition:

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

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

The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5


解法1:利用归并排序排序思想(Accepted)
由于楼主比较笨,所以第一时间能想到的方法就是利用归并排序,虽然时间复杂度没有做到 O(log(m+n)) O ( l o g ( m + n ) ) ,但是依然 AC 了。这种想法很好实现,只不过需要注意的是边界和奇偶的处理。程序中,我并没有真正去建立一个大数组存储归并之后的所有元素,只是使用 pnums1 和 pnums2 分别记录两个数组归并到了哪里,使用 p 记录已经归并的元素的个数,low 在两数组和为奇数时就是中位数,在两数组和为偶数时是低中位数,这时继续寻找下一个即将归并到大数组中的元素作为高中位数,然后取平均。

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        n = len(nums1) + len(nums2)
        p = n /2 if n % 2 == 0 else (n + 1)/2
        low = 0
        pnums1 = pnums2 = 0
        while p != 0:
            if pnums1 == len(nums1):
                low = nums2[pnums2]
                pnums2 += 1
            elif pnums2 == len(nums2):
                low = nums1[pnums1]
                pnums1 += 1
            elif nums1[pnums1] > nums2[pnums2]:
                low = nums2[pnums2]
                pnums2 += 1
            else:
                low = nums1[pnums1]
                pnums1 += 1
            p -= 1
        if n % 2 == 0:
            if pnums1 == len(nums1):
                high = nums2[pnums2]
            elif pnums2 == len(nums2):
                high = nums1[pnums1]
            else:
                high = nums1[pnums1] if nums1[pnums1] < nums2[pnums2] else nums2[pnums2]
            return float(low + high) / 2
        else:
            return low

解法2:二分法(Accepted)
在此题的 Solution 部分已经将这种方法解释的很明白了,并且有 Java 和 Python 的实现,因此这里我就不再说了。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值