[백준] 2491 - 수열(python)

https://www.acmicpc.net/problem/2491

 

문제

0에서부터 9까지의 숫자로 이루어진 N개의 숫자가 나열된 수열이 있다. 그 수열 안에서 연속해서 커지거나(같은 것 포함), 혹은 연속해서 작아지는(같은 것 포함) 수열 중 가장 길이가 긴 것을 찾아내어 그 길이를 출력하는 프로그램을 작성하라. 

예를 들어 수열 1, 2, 2, 4, 4, 5, 7, 7, 2 의 경우에는 1 ≤ 2 ≤ 2 ≤ 4 ≤ 4 ≤ 5 ≤ 7 ≤ 7 이 가장 긴 구간이 되므로 그 길이 8을 출력한다. 수열 4, 1, 3, 3, 2, 2, 9, 2, 3 의 경우에는 3 ≥ 3 ≥ 2 ≥ 2 가 가장 긴 구간이 되므로 그 길이 4를 출력한다. 또 1, 5, 3, 6, 4, 7, 1, 3, 2, 9, 5 의 경우에는 연속해서 커지거나 작아지는 수열의 길이가 3 이상인 경우가 없으므로 2를 출력하여야 한다.

 

입력

 

첫째 줄에는 수열의 길이 N이 주어지고, 둘째 줄에는 N개의 숫자가 빈칸을 사이에 두고 주어진다. N은 1 이상 100,000 이하의 정수이다.

출력

첫째 줄에 가장 긴 길이를 출력한다.

---

소스코드

n = int(input())
li = list(map(int, input().split()))
dp = [0 for i in range(n+1)]
dp[0] = 1

p = 1
m = 1
for i in range(1, n):
    if li[i] == li[i-1]:
        p += 1
        m += 1
    elif li[i] > li[i-1]:
        m = 1
        p += 1
    else:
        p = 1
        m += 1

    dp[i] = max(p, m)

print(max(dp))

 

소스코드 설명

1. 리스트 li[i]가 li[i-1]와 값이 같을 경우 변수 p, m을 각각 1씩 더한다.

2. 리스트 li[i]가 li[i-1]보다 클 경우 변수 p에 1을 더하고, m 은 1로 설정한다.

3. 리스트 li[i]가 li[i-1]보다 작을 경우 변수 m에 1을 더하고, p는 1로 설정한다.

4. dp[i]는 p, m중 큰 값을 지정한다.

5. dp 리스트중 가장 큰 값을 출력한다.