백준 파이썬 공부 2025.10.25
2096번 내려가기 (골드 5)
1. 문제
N줄에 0 이상 9 이하의 숫자가 세 개씩 적혀 있다.
내려가기 게임을 하고 있는데, 이 게임은 첫 줄에서 시작해서 마지막 줄에서 끝나게 되는 놀이이다.
먼저 처음에 적혀 있는 세 개의 숫자 중에서 하나를 골라서 시작하게 된다.
그리고 다음 줄로 내려가는데, 다음 줄로 내려갈 때에는 다음과 같은 제약 조건이 있다.
바로 아래의 수로 넘어가거나, 아니면 바로 아래의 수와 붙어 있는 수로만 이동할 수 있다는 것이다.
이 제약 조건을 그림으로 나타내어 보면 다음과 같다.

별표는 현재 위치이고, 그 아랫 줄의 파란 동그라미는 원룡이가 다음 줄로 내려갈 수 있는 위치이며,
빨간 가위표는 원룡이가 내려갈 수 없는 위치가 된다.
숫자표가 주어져 있을 때, 얻을 수 있는 최대 점수, 최소 점수를 구하는 프로그램을 작성하시오.
점수는 원룡이가 위치한 곳의 수의 합이다.
2. 입력
첫째 줄에 N(1 ≤ N ≤ 100,000)이 주어진다.
다음 N개의 줄에는 숫자가 세 개씩 주어진다.
숫자는 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 중의 하나가 된다.
3. 출력
첫째 줄에 얻을 수 있는 최대 점수와 최소 점수를 띄어서 출력한다.
4. 문제 풀이
다이나믹 프로그래밍 문제이다. 최소값과 최대값 모두를 구하기 위해 DP를 두 개 시행해야한다.
>>코드1. 메모리 초과
n = int(input())
score = [list(map(int, input().split())) for _ in range(n)]
DP1 = list([0] * (n+2) for _ in range(n+1))
DP2 = list([100001] * (n+2) for _ in range(n+1))
DP2[0] = [0] * (n+2)
for i in range(1, n+1):
for j in range(1, n+1):
DP1[i][j] = max(DP1[i-1][j-1], DP1[i-1][j], DP1[i-1][j+1]) + score[i-1][j-1]
DP2[i][j] = min(DP2[i-1][j-1], DP2[i-1][j], DP2[i-1][j+1]) + score[i-1][j-1]
print(max(DP1[-1]), min(DP2[-1]))
문제 해결을 위해 각 자리에서의 최대값과 최소값을 저장하는
(n+2) X (n+1) 리스트를 생성했으나 메모리 초과가 발생하였다.
이전 경로들의 최대값과 최소값은 저장할 필요가 없으니 일차원 리스트에 다시 최대값과 최소값을 저장해보자.
>> 정답코드
"""
2096번 내려가기
입력 : n (층수), score (위치의 점수)
출력 : max_score (최대 점수), min_score (최소 점수)
(이동 경로의 모든 점수의 합의 최댓값과 최솟값)
"""
n = int(input())
a, b, c = map(int, input().split())
max_dp = [a, b, c]
min_dp = [a, b, c]
for _ in range(n - 1):
a, b, c = map(int, input().split())
prev_max = max_dp[:]
prev_min = min_dp[:]
max_dp[0] = a + max(prev_max[0], prev_max[1])
max_dp[1] = b + max(prev_max[0], prev_max[1], prev_max[2])
max_dp[2] = c + max(prev_max[1], prev_max[2])
min_dp[0] = a + min(prev_min[0], prev_min[1])
min_dp[1] = b + min(prev_min[0], prev_min[1], prev_min[2])
min_dp[2] = c + min(prev_min[1], prev_min[2])
print(max(max_dp), min(min_dp))
일차원 리스트에 매 층의 위치에서의 최소경로와 최대경로 값을 저장하여,
각각 길이 n의 일차원 리스트만 필요했다.
5. 문제 링크
https://www.acmicpc.net/problem/2096
'백준 > 백준 파이썬' 카테고리의 다른 글
| 백준 파이썬 1504번 특정한 최단 경로 (Today I Learn 2025.10.27) (0) | 2025.10.29 |
|---|---|
| 백준 파이썬 15686번 치킨 배달 (Today I Learn 2025.10.26) (0) | 2025.10.28 |
| 백준 파이썬 9251번 LCS (Today I Learn 2025.10.23) (0) | 2025.10.27 |
| 백준 파이썬 12865번 평범한 배낭 (Today I Learn 2025.10.15) (0) | 2025.10.25 |
| 백준 파이썬 13549번 숨바꼭질 3 (Today I Learn 2025.10.14) (0) | 2025.10.16 |
