📊알고리즘, 문제풀이/📈문제풀이 (PS)

[알고리즘][Python] 백준 1149 RGB 거리 문제 풀이

Written by Donghak Park

문제 출처 : www.acmicpc.net/problem/1149

 

1149번: RGB거리

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

www.acmicpc.net


문제 해석 : RGB 거리를 구하는 문제로 각각 RGB마다 주어진 값이 있다. 이 값을 최소로 만들어야 한다. 이때 자신의 바로 앞에서 선택된 색은 고를 수 없다.

 

문제 풀이 : DP로 문제를 해결 할 수 있다. -> 각 3가지 경우 모두를 매번 자신의 색을 제외한 최소값으로 고르면서 진행하면 마지막에 최소를 구할 수 있다.


풀이 코드

import sys
input = sys.stdin.readline

def solution():
    global arr

    for i in range(1, N):
        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]

    return min(arr[N-1][0], arr[N-1][1], arr[N-1][2])

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

print(solution())

author : donghak park
contact : donghark03@naver.com

## 문제의 저작권은 백준 알고리즘 사이트에 있습니다. 혹시 문제가 되는 부분이 있으면 연락 바랍니다.