[Python] 백준(BOJ) 2491번: 수열

문제

2491번: 수열

풀이

올라가는게 연속되는경우, 내려가는게 연속되는 경우를 is_increasing으로 관리하면서 풀었는데

같은 값이 반복되는 경우의 반례에서 걸렸다.

예를들어 5 4 4 2 2나 1 2 2 3 3 같은 경우도 테스트해서 풀어야함

그래서 same 변수로 몇번 반복되었는지도 관리해줬다. 

import sys

N = int(sys.stdin.readline())
li = list(map(int, sys.stdin.readline().split(" ")))

result = 0
is_increasing = True
count = 1
same = 0
for i in range(N-1):
    if li[i] < li[i+1]:
        if is_increasing:
            count += 1
        else:
            is_increasing = True
            if count > result:
                result = count
            count = same + 2
        same = 0
    elif li[i] > li[i+1]:
        if is_increasing:
            is_increasing = False
            if count > result:
                result = count
            count = same + 2
        else:
            count += 1
        same = 0
    else:
        count += 1
        same += 1

print(max(result, count))