【LeeetCode】4. 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)).

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

题意：找出两个数组的中间值

思路：首先是想利用像归并排序的方法，排序完成后，选出中间值，但是时间复杂度就不是O(log(m+n))了

从网上找到另一种方法，我写下我自己理解的思路：

找出中间值，也就是找出第K小的数

　　借用二分查找法的思路，第k/2小的值在或者不在短的数组中，就比如短的数组事nums1。

　　如果k/2在短的数组中，就比较nums1[k/2]和nums2[k/2]，比如如果nums1[k/2]更小，那么就可以排除掉nums1中，索引k/2之前的所有数。

　　如果k/2不在短的数组中，就比较nums1[len(nums1)]和nums2[k-len(nums1)]的值，谁更小，就排除掉哪个数组

　　如果两个值相等，就找到了！

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        len1 = len(nums1)
        len2 = len(nums2)
        l = len1 + len2
        if l&0x1 == 1:
            return self.findK(nums1,len1,nums2,len2,l/2+1)
        else:
            return (self.findK(nums1,len1,nums2,len2,l/2)+self.findK(nums1,len1,nums2,len2,l/2+1))*1.0/2
    
    def findK(self,nums1,len1,nums2,len2,k):
        if len1>len2:
            return self.findK(nums2,len2,nums1,len1,k)
        if len1 == 0:
            return nums2[k-1]
        if k == 1:
            return min(nums1[0],nums2[0])
        mid = k/2
        pa = min(mid,len1)
        pb = k-pa
        if nums1[pa-1] < nums2[pb-1]:
            return self.findK(nums1[pa:],len1-pa,nums2,len2,k-pa)
        elif nums1[pa-1] > nums2[pb-1]:
            return self.findK(nums1,len1,nums2[pb:],len2-pb,k-pb)
        else:
            return nums1[pa-1]