백준(Python) 풀이/다이나믹 프로그래밍

백준(Python) 1149번 RGB거리 풀이

개발윗미 2022. 9. 16. 09:02

Python으로 구현한 1149번 RGB거리 문제 풀이입니다.

 

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

 

1149번: RGB거리

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

www.acmicpc.net


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

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

print(min(data[n-1][0], data[n-1][1], data[n-1][2]))

 

1. 다이나믹 프로그래밍을 통해 문제를 해결할 수 있다.

 

2. 1부터 n까지를 반복문의 범위로 설정하여 각 data[i][x]에 이전 좌표에서 가능한 인덱스의 최소값과 현재 좌표의 최소값을 더하여 갱신한다.

 

3. 최종적으로 마지막 인덱스의 0번째, 1번째, 2번째 값 중 최솟값을 출력한다.