LeetCode 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

整体思路

因为题目中对时间复杂度要求是log(m+n)级别的，直接想法是：

递归：大问题转化为小问题。
二分查找：利用数组的有序性。

递归首先需要定义问题，开始想到的是：获取两个升序数组的中间值。
对于两数组总体元素为奇数的情况，这种问题定义是满足的。
但是对于偶数的情况，这种问题定义略微复杂。

所以，转化问题定义：获取两个个升序列数组中，整体从大到小排序后，第targetpos个元素。

对于奇数总数情况，寻找的是第(m + n)/2 + 1个。
对于偶数总数情况，寻找的是第(m + n)/2和(m + n)/2 +1 个的平均，需要求两次。

1、递归边界

根据递归，首先想到的是问题的边界，也就是不再递归的情况：
1、两个数组中存在空数组或者单一元素数组。
2、两个升序数组，一个数组的最小值大于另外一个最大值。
3、targetpos为1。

2、递归拆解

开始想的是每次拆分一个数组的一半，把某个数组第一个元素拿出来，通过在第二个数组中二分查找，看看在第二个数组中能排第几个。
然后，把小于该元素在第二个数组中的元素都删除，可以进行问题的拆分。

后来想想，这里没用到第一个数组的有序性，而且也没对第一个数组进行问题的拆解，所以继续进行优化：

把第一个数组的中间值取出来，放在第二个数组中，利用二分查找，看看在第二个数组中能排第几个，这样，不论第一个数组中间值大了或者小了，都能每次对两个数组整体进行一半拆分。

3、复杂度分析

因为每次对(m+n)的整体进行拆分，所以整体递归复杂度O(log(m+n))。
二分查找第一个数组中间值在第二个数组中排第几，需要用二分查找，复杂度O(log(n))。
整体复杂度O(log(m+n) + log(n))，其实是满足题目要求的。

Python代码

# coding: utf-8
class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        l1 = len(nums1)
        l2 = len(nums2)

        if