LeetCode findMedianSortedArrays Python3

题目描述：给定两个大小为 m 和 n 的有序数组 A 和 B 。找出这两个有序数组的中位数，并且要求算法时间复杂度为 O(log(m+n))。可以假设这两个数组不会同时为空。
示例 1：
输入： A = [1,3], B = [2]
输出： 2.0
示例 2：
输入： A = [1,2], B = [3,4]
输出： (2+3) / 2 = 2.5

其实如果没有时间复杂度的要求，这个问题用 Python 实在是太简单了，感觉有点像作弊：

def findMedianSortedArrays(A, B):
	res = []
	res.expand(A)
	res.expand(B)
	res = sorted(res)
	if len(res) % 2 == 0:
		half = len(res) // 2
		return (res[half-1] + res[half]) / 2.0
	else:
		return res[len(res)//2] / 1.0

但是这个代码在效率上是不合格的，因为只是使用 sorted 函数 O 就已经爆掉了。看到给出的示例大概是这样的：

def findMedianSortedArrays(A, B):
	m, n = len(A), len(B)
	if m > n:
		A, B, m, n = B, A