(Python)백준 코딩테스트 연습

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

HRuler 2023. 6. 28. 12:49

1. 문제

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

1149번: RGB거리

첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나

www.acmicpc.net

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]))