打印极值点坐标

#将数组分成三部分,第一部分是中间的,第二和第三分别是头两个数和最后两个数。

#分别找出这三个部分的极值点,合起来就可以了。

def filterPeak(arr):
    a = list(
        filter(lambda i: (arr[i] < arr[i - 1] and arr[i] < arr[i + 1]) or (arr[i] > arr[i - 1] and arr[i] > arr[i + 1]),
               range(1, len(arr) - 1)))
    b = [0] if arr[0] != arr[1] else []
    c = [len(arr) - 1] if arr[-2] != arr[-1] else []
    return b + a + c
 
a, b = input(), list(map(int, input().split()))
print(" ".join(map(str, filterPeak(b))))

猜你喜欢

转载自blog.csdn.net/qq_29380377/article/details/89438418