Пилообразная последовательность

text = open("input.txt");
length = int(text.readline());
arr = text.readline();
text2 = open("output.txt", 'w');
arr = arr.split(" ")
i = 1;
k = 1;
j = 1;
l = 1;
n = 0;
a2 = []
a3 = []
a5 = []
d = 0 # итог
m = [] # индексы нулей
indexes = []
while i < length:
    if arr[i-1] < arr[i]:
        a2.append(-1)
    if arr[i-1] > arr[i]:
        a2.append(1)
    if arr[i-1] == arr[i]:
        a2.append(0)
    i = i + 1;
while k < len(a2):
    if (a2[k-1] == -1 and a2[k] == 1) or (a2[k-1] == 1 and a2[k] == -1):
        a3.append(1)
    else:
        a3.append(0)
    k = k + 1
if length == 1:
    d = 1
if a3.count(1) == 0 and length > 1:
    d = 2
if a3.count(1) == 1 and length > 1:
    d = 3
if a3.count(0) == 0 and length > 1:
    d = len(a3) + 2
if a3.count(0) > 0 and a3.count(1) > 1:
    a3.append(0)
    a3.reverse()
    a3.append(0)
    a3.reverse()
    # вставляем нули в начало и конец списка на всякий случай
    while j < len(a3):
        indexes.append(a3.index(0, j-1, len(a3)))
        j = j + 1
    indexes = sorted(indexes)
    i = 0
    while l < len(indexes):
        a5.append(indexes[l] - indexes[l - 1])
        l = l + 1
    while i < len(a5):
        if(a5[i] > n):
            n = a5[i]
        i = i + 1
    d = n + 1
 
text2.write(str(d))
text.close()
text2.close();

def solve(a):
    return sum(1 for x,y,z in zip(a, a[1:], a[2:]) if x < y > z or x > y < z) + 2
 
a = list(map(int, '9 8 7 -5 3 1 4 7 6'.split()))
print(solve(a))