码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode 解题 Median of Two sorted arrays

时间:2015-04-30 23:08:07      阅读:128      评论:0      收藏:0      [点我收藏+]

标签:

题目: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))

题解:

1、自己想得思路:构建一个list,然后比较各数的大小,将其插入到合适的位置

class Solution:
    # @param {integer[]} nums1
    # @param {integer[]} nums2
    # @return {float}
    def findMedianSortedArrays(self, nums1, nums2):
        nums = nums1[:]    # 构建一个list,并将nums1的值赋给它
        x = len(nums1)
        y = len(nums2)
        for i in range(x + y):
            if i < len(nums):
                if len(nums2) == 0: break    # 如果nums2没有数了就跳出
                elif nums[i] < nums2[0]:
                    continue
                else:    # 否则将nums2[0]插入到i位置
                    num = nums2.pop(0)
                    nums.insert(i, num)
            else: break
        nums.extend(nums2)
        n = len(nums)/2        # 输出结果
        if len(nums)%2 == 0: return (nums[n] + nums[n-1])/2.0
        else: return nums[n]

 2、参考网上的解题思路(http://c4fun.cn/blog/2014/03/20/leetcode-solution-02/)

用求两个有序数组的第K大数的方法:

假设A数组中取第X个数, B数组中取第Y个数,并且满足X+Y=K, 若A[X] < B[Y],则比A[X]小的数必然少于K个,也就是说A[1]到A[X]都比第K个数要小,可以舍弃掉然后求第K-X小的数,反之亦然

class Solution:
    def findMedianSortedArrays(self, A, B):
        totlen = len(A) + len(B)
        if (1 & totlen):    # 通过位运算判断奇偶数,nice
            return self.findK(A, B, (totlen+1)/2)
        else:
            return (self,findK(A, B, totlen/2) + self.findK(A, B, totlen/2+1))/2.0

    def findK(self, A, B, K):
        la, lb, pa, pb = len(A), len(B), min(K/2, len(A)), K - (min(K/2, len(A)))
        if (la > lb): return self.findK(B, A, K)
        if (la == 0): return B[K-1]
        if (K == 1): return min(A[0], B[0])
        if A[pa-1] < B[pb-1]: return self.findK(A[pa:], B, K-pa)
        elif A[pa-1] > B[pb-1]: return self.findK(A, B[pb:], K-pb)
        else: return A[pa-1]

 

Leetcode 解题 Median of Two sorted arrays

标签:

原文地址:http://www.cnblogs.com/siriuswang/p/4469987.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!