【题目】
给定两个有序数组arr1和arr2,再给定一个整数k,返回所有的数中第k小的数。要求时间复杂度O(log(min{M,N})),额外空间复杂度O(1)。
【基本思路】
在解决这道题之前,先解决一个小问题:在两个长度相等的排序数组中找到上中位数。本题也深度利用了这个问题的解法。以下的getUpMedian方法的功能就是,在a1[s1…e1]和a2[s2…e2]两段长度相等的范围上找上中位数。
def getUpMedian(a1, s1, e1, a2, s2, e2):
while s1 < e1:
mid1 = (e1 + s1) // 2
mid2 = (e2 + s2) // 2
offset = (e1 - s1 + 1) & 1