[백준-1149] RGB거리 / Python

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

1149번: RGB거리

n = int(input())
data = []
for _ in range(n):
    data.append(list(map(int, input().split())))

for i in range(1, n):
    for j in range(3):
        if j == 0:
            data[i][j] = min(data[i - 1][1], data[i - 1][2]) + data[i][j]
        elif j == 1:
            data[i][j] = min(data[i - 1][0], data[i - 1][2]) + data[i][j]
        elif j == 2:
            data[i][j] = min(data[i - 1][0], data[i - 1][1]) + data[i][j]

print(min(data[-1]))

 

해설

간단한 문제였다. 이전 집의 자기와 다른 인덱스를 가진 값의 최소를 더해주면 된다.