等长排序数组的中位数

Question

我正在尝试找到两个相同长度的排序数组的中位数，我正在使用分治算法。但是我的代码返回 None 而不是一个值 这是查找单个数组中位数的代码：

def getmedian(li):
    x = len(li)
    if x%2 == 0:
        return (li[x//2] + li[(x//2)-1])/2
    else:
        return li[(len(li)-1)//2]

然后我对两个数组使用以下函数：

def TwoList(list1,list2):
    if len(list1) == 1:
        return (list1[0] + list2[0])/2
    elif len(list2)== 2:
        return (max(list1[0],list2[0]) + min(list2[1], list1[1]))/2
    else:
        #pdb.set_trace()
        x = getmedian(list1)
        y = getmedian(list2)
        if x == y:
            return x
        elif x > y:
            if len(list1)%2==0:
                TwoList(list1[:len(list1)//2], list2[len(list1)//2:])
            else:
                TwoList(list1[:len(list1)//2+1], list2[len(list1)//2:])
        else:
            if len(list1)%2==0:
                TwoList(list1[len(list1)//2:], list2[:len(list1)//2])
            else:
                TwoList(list1[len(list1)//2:], list2[:len(list1)//2+1])

Answer 1

看起来你做对了，除了在TwoList递归调用中返回值。
请看下面的代码，在你必须return值的地方注释。

def getmedian(li):
    x = len(li)
    if x%2 == 0:
        return (li[x//2] + li[(x//2)-1])/2
    else:
        return li[(len(li)-1)//2]

def TwoList(list1,list2):
    if len(list1) == 1:
        return (list1[0] + list2[0])/2
    elif len(list2)== 2:
        return (max(list1[0],list2[0]) + min(list2[1], list1[1]))/2
    else:
        #pdb.set_trace()
        x = getmedian(list1)
        y = getmedian(list2)
        if x == y:
            return x
        elif x > y:
            if len(list1)%2==0:
                return TwoList(list1[:len(list1)//2], list2[len(list1)//2:]) # return value
            else:
                return TwoList(list1[:len(list1)//2+1], list2[len(list1)//2:]) # return value
        else:
            if len(list1)%2==0:
                return TwoList(list1[len(list1)//2:], list2[:len(list1)//2]) # return value
            else:
                return TwoList(list1[len(list1)//2:], list2[:len(list1)//2+1]) # return value

print(TwoList([1,2,3,4],[5,6,7,8]))