(Python)백준 코딩테스트 연습 - RGB거리(1149)

1. 문제

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

1149번: RGB거리

2. 풀이

 

import sys
input = sys.stdin.readline

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

for i_k, i_v in enumerate(rgbList[1:]):
    tempList = []
    for j_k, j_v in enumerate(i_v):
        if j_k == 0:
            tempList.append(j_v + min(dp[i_k][1], dp[i_k][2]))
        elif j_k == 1:
            tempList.append(j_v + min(dp[i_k][0], dp[i_k][2]))
        elif j_k == 2:
            tempList.append(j_v + min(dp[i_k][0], dp[i_k][1]))
    dp.append(tempList)

print(min(dp[-1]))