본문 바로가기
카테고리 없음

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

by 육빔 2024. 3. 28.
728x90

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

 

1149번: RGB거리

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

www.acmicpc.net

 

쉽지 않았던 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]))

 

완성