寻找两个有序数组中的第K个数

方法一：双指针

两个指针各指在两个数组的开头，谁小移谁，移动k次就能找到。时间复杂度为O(k)

方法二：分治法

解释在代码中

# time: log(m+n)
def FindKthElm(a, b, k):
    if len(a) == 0:
        return b[k-1]
    if len(b) == 0:
        return a[k-1]
    
    a_mid = len(a) // 2
    b_mid = len(b) // 2
    half_len = a_mid + b_mid + 2 # a,b数组前半部分（包括Mid）的大小
    
    if a[a_mid] < b[b_mid]:
        if half_len > k:
            # K（K指的是第k大的数）这个数在合并数组内
            # 因为b[b_mid]必定是合并数组中最大的那个，那么b[b_mid]一定比K的数大
            # 所以b[b_mid~end]的数就不用搜索了，因为它们必定比K大
            return FindKthElm(a[:], b[:b_mid], k)
#         elif half_len == k: # 不能用==去判断
#             # K就是b[b_mid]
#             return b[b_mid]
        else:
            # 此时在合并的数组中a[:a_mid+1]元素一定在b[b_mid]的左侧，
            # 所以前K个元素中一定包含A[:a_mid+1]（可以使用反证法来证明这点）
            # 但是无法判断A[a_mid+1:]与B[:]之间的关系，需要对他们进行判断
            return FindKthElm(a[a_mid+1:], b[:], k-(a_mid+1))
    else:
        if half_len > k:
            # 同上
            return FindKthElm(a[:a_mid], b[:], k)
#         elif half_len == k:
#             return a[a_mid]
        else:
            return FindKthElm(a[:], b[b_mid+1:], k-(b_mid+1))
        
a = [1, 6, 8, 9, 11]
b = [2, 4, 5, 7, 10]
FindKthElm(a, b, 6)