[백준/1149] RGB거리 (with 파이썬)

✔ 문제

---

✔ 풀이

🔥 index error

import sys
input = sys.stdin.readline

n = int(input())
rgb = []

for _ in range(n):
    rgb.append(list(map(int, input().split())))

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

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

 

👾 정답

import sys
input = sys.stdin.readline

n = int(input())
rgb = []

for _ in range(n):
    rgb.append(list(map(int, input().split())))

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

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

---

✔ 설명

맨 처음 집은 for문에 포함하지 않고 두번째 집부터 포함한다.

현재 [i][0] (빨간집) 인 경우 전 집의 초록집, 파란집 둘 중에서 더 작은 집과 현재의 집의 가격을 더한 값을 [i][0]에 대입한다.

계속하여 값을 누적하기 때문에 rgb[n-1] 값 중 가장 작은 값을 출력하면 된다.

---

✔ 후기

현재 집을 기준으로 i-1, i+1 값도 같이 더해서 현재 값을 저장하는 줄 알고 구현했다가 index error가 나서 다시 수정했다.