PS/Python
[Python] 백준(BOJ) 2491번: 수열
yoo.o
2021. 11. 10. 16:37
반응형
문제
2491번: 수열
0에서부터 9까지의 숫자로 이루어진 N개의 숫자가 나열된 수열이 있다. 그 수열 안에서 연속해서 커지거나(같은 것 포함), 혹은 연속해서 작아지는(같은 것 포함) 수열 중 가장 길이가 긴 것을 찾
www.acmicpc.net
풀이
올라가는게 연속되는경우, 내려가는게 연속되는 경우를 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))
반응형