난이도: 1.5 / 풀이 시간: 30분
n x m 크기의 금광이 있습니다. 금광은 1 x 1 크기의 칸으로 나누어져 있으며, 각 칸은 특정한 크기의 금이 들어 있습니다.
채굴자는 첫 번째 열부터 출발하여 금을 캐기 시작합니다. 맨 처음에는 첫 번째 열의 어느 행에서든 출발할 수 있습니다.
이후에 m번에 걸쳐서 매번 오른쪽 위, 오른쪽, 오른 쪽 아래 3가지 중 하나의 위치로 이동해야 합니다.
결과적으로 채굴자가 얻을 수 있는 금의 최대 크 기를 출력하는 프로그램을 작성하세요.
만약 다음과 같이 3 x 4 크기의 금광이 존재한다고 가정합시다.
1 3 3 2 2 1 4 1 0 6 4 7
가장 왼쪽 위의 위치를 (1, 1), 가장 오른쪽 아래의 위치를 (n, m)이라고 할 때, 위 예시에서는 (2, 1) -> (3,2) -> (3, 3) -> (3, 4)의 위치로 이동하면 총 19만큼의 금을 채굴할 수 있으며, 이때의 값이 최댓값입니다.
입력조건
출력 조건
입력 예시
2 3 4 1 3 3 2 2 1 4 1 0 6 4 7 4 4 1 3 1 5 2 2 4 1 5 0 2 3 0 6 1 2
출력 예시
19
16
풀이 특징
import sys
import copy
input = sys.stdin.readline
t = int(input())
for _ in range(t):
n, m = map(int, input().split())
array = list(map(int, input().split()))
index = 0
graph = []
for i in range(n):
graph.append(array[index:index+m])
index += m
# DP 시작
dp = copy.deepcopy(graph)
maxValue = 0
for j in range(1, m):
for i in range(n):
dp[i][j] = max(dp[i][j], dp[i][j-1] + graph[i][j])
if i-1 >= 0:
dp[i][j] = max(dp[i][j], dp[i-1][j-1] + graph[i][j])
if i+1 < n:
dp[i][j] = max(dp[i][j], dp[i+1][j-1] + graph[i][j])
maxValue = max(maxValue, dp[i][j])
for i in range(n):
for j in range(m):
print(dp[i][j], end=" ")
print()
print()
print(maxValue)