백준 1149번 RGB거리 파이썬 (python code)

난이도: 실버1
번호: 1149번

www.acmicpc.net/problem/1149

1149번: RGB거리

DP 문제인데 이것도 그리디한 방식으로 접근했다가 다시 돌아왔다.

다이나믹프로그래밍이 아직 익숙하지 않은것같다.

각 줄에 대해 최소합으로 최신화시키면서 나아가고 마지막에 min을 취해 결과값을 얻었다.

from sys import stdin

N = int(stdin.readline().rstrip())
paint=[0]*N

for i in range(0,N):
    paint[i] = list(map(int,stdin.readline().split()))

for i in range(1, N):
    paint[i][0] = min(paint[i - 1][1], paint[i - 1][2]) + paint[i][0]
    paint[i][1] = min(paint[i - 1][0], paint[i - 1][2]) + paint[i][1]
    paint[i][2] = min(paint[i - 1][0], paint[i - 1][1]) + paint[i][2]
print(min(paint[N - 1][0], paint[N - 1][1], paint[N - 1][2]))