【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]