[백준/파이썬] 1149번

📍백준 1149 문제

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

코드

import sys
input = sys.stdin.readline

n = int(input())

color = []
for _ in range(n):
    color.append(list(map(int, input().split())))
for i in range(1, n):
    color[i][0] = min(color[i-1][1], color[i-1][2])+color[i][0]
    color[i][1] = min(color[i-1][0], color[i-1][2])+color[i][1]
    color[i][2] = min(color[i-1][1], color[i-1][0])+color[i][2]

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

풀이

• 색을 기준으로 문제를 풀어준다.
• 현재 위치가 빨간색이라면 이전 값에서 파란색, 초록색 중 최소값을 뽑아 현재 값에 더해준다.
  -> 다른 색도 마찬가지로 하면 된다.