[백준_1149] RGB 거리

문제 보기: https://www.acmicpc.net/problem/1149

1149번: RGB거리

문제

RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.

집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.

• 1번 집의 색은 2번 집의 색과 같지 않아야 한다.
• N번 집의 색은 N-1번 집의 색과 같지 않아야 한다.
• i(2 ≤ i ≤ N-1)번 집의 색은 i-1번, i+1번 집의 색과 같지 않아야 한다.

 

입력

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

출력

첫째 줄에 모든 집을 칠하는 비용의 최솟값을 출력한다.

풀이 과정

1. 해당 문제의 규칙을 생각해본다 색을 칠하는데 이전 순서와 같은 색을 칠하지 못하고 남은 두개 숫자중에 작은것을 구하는 것으로 생각하고 시작한다.

2. 배열을 이용하여 입력값을 입력 받도록 하며 1번의 식을 점화식으로 만들어서 사용한다

3. 마지막 순서에는 중간 단계에서 하는 방식과 똑같이 하여 마지막 값을 뽑아내면 끝

 

코드

def solve():
    n = int(input())
    rst = []
    r, g, b = [int(x) for x in input().split()]
    rst.append((r, g, b))
    for _ in range(1, n):
        r, g, b = [int(x) for x in input().split()]
        pR, pG, pB = rst[-1]
        R = r + min(pG, pB)
        G = g + min(pR, pB)
        B = b + min(pR, pG)
        rst.append((R, G, B))
    fR, fG, fB = rst[-1]
    print(min(fR, fG, fB))


solve()