[백준] 1149번: RGB거리 - python

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

1149번: RGB거리

 

쉽지 않았던 dp문제.

 

값을 업데이트 해나가는 방식이 접근하기 어려웠다. 최소값은 전에 있는 것만 비교하면 되기 때문에 i-1과 비교하여 그 자리를 업데이트한다.

그러면 각 R G B에 대한 최소값이 업데이트 되는데 마지막에 그 3가지 중 최소값을 찾으면 되는 문제이다.

n = int(input())

arr = []
for i in range(n):
    R, G, B = map(int, input().split())
    arr.append([R, G, B])

for i in range(1, n): 
    #앞에 있는 것과 비교한 후 R, G, B를 업데이트 시킨다.
    arr[i][0] = min(arr[i - 1][1], arr[i - 1][2]) + arr[i][0]
    arr[i][1] = min(arr[i - 1][0], arr[i - 1][2]) + arr[i][1]
    arr[i][2] = min(arr[i - 1][0], arr[i - 1][1]) + arr[i][2]

print(min(arr[n - 1]))

 

완성