挑出峰值数字和所处位置（coderwars 5级挑战题）

ivor

看着还不错，和大家分享一下。


名称：挑出峰值数字和所处位置

详细介绍：

例子1, 数组 arr = [0, 1, 2, 5, 1, 0] 峰值索引位置在 3，峰值对应为 5 (since arr[3] equals 5). 结果：{pos: [3], peaks: [5]}
例子2，数组 arr = [1,2,5,4,3,2,3,6,4,1,2,3,3,4,5,3,2,1,2,3,5,5,4,3]), 索引位置在[2,7,14,20] 峰值对应为[5,6,5,5]  结果 ：{"pos":[2,7,14,20], "peaks":[5,6,5,5]})


抛砖引玉附上Python代码:

a = [11, 20, 14, 4, 5, 4, 8, 17, 14, 9, 17, 20, 8, 19, 17, 15, 13, 20, 16, 7, 2, 11, 14, 18, 11, 2, 15, 1, 5, 4, 5, 1, 6, 6, 9, 13, 18, 19, 9, 17, 12, 19, 17, 2, 13, 9, 19, 2, 11]
def pick_peaks(arr):
    pos = []
    prob_peak = False
    for i in range(1, len(arr)):
        if arr[i] > arr[i-1]:
            prob_peak = i
        elif arr[i] < arr[i-1] and prob_peak:
            pos.append(prob_peak)
            prob_peak = False
    return {'pos':pos, 'peaks':[arr[i] for i in pos]}
print(pick_peaks(a))
复制代码

ivor

回复 2# Gin_Q


    比如：例子2索引位置14比左右数值都小，如果和右边数值相等，则继续往后找，找到后才能确定14位置为峰值，如果右边找不到更小的数值则不成立。

形象比喻就像山头。

ivor

回复 5# xczxczxcz


    忽略了一种相邻两个(或更多个)数可能是相等的情况。