Leetcode 解题 Median of Two sorted arrays

题目：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]