[Python] 寻找局部最高点

[Python] 寻找局部最高点

# 寻找局部最高点
# 输入input: 含有最高点高度的列表
# 输出output: 返回最高点的位置
# 时间复杂度: O(log(n))

def findHighSpot(input):
    iLen = len(input)
    mid = iLen//2 - 1 # Python中'/'为小数除法，'//'为地板除
    if input[mid-1] <= input[mid] >= input[mid+1]:
        return str(mid+1) # 默认返回最高点到第一个点之间的距离
    elif input[mid-1] > input[mid]:
        return findHighSpot(input[:mid//2])
    else:
        return 'mid+'+findHighSpot(input[mid//2:]) # 返回最高点距离中间点的距离

if __name__ == '__main__':
    input = [1, 2, 6, 5, 3, 7, 4]
    output = findHighSpot(input)
    print('output:', output) # output: 3