RGB거리2 17404번 백준

문제:www.acmicpc.net/problem/17404

17404번: RGB거리 2

 

import sys
n = int(input())
m = []
for _ in range(n):
    m.append([int(i) for i in sys.stdin.readline().split()])
 
ret = 1001*1001
 
for x in range(3): 
    dp = [[0 for _ in range(n)] for _ in range(3)]
 
    for i in range(3): 
        if i == x:
            dp[i][0] = m[0][i]
            continue
        dp[i][0] = 1001*1001
 
    for i in range(1,n): 
        dp[0][i] = m[i][0]+min(dp[1][i-1],dp[2][i-1])
        dp[1][i] = m[i][1]+min(dp[0][i-1],dp[2][i-1])
        dp[2][i] = m[i][2]+min(dp[0][i-1],dp[1][i-1])
    
    for i in range(3): 
        if i == x:
            continue
        ret = min(ret, dp[i][-1])
print(ret)​

참고: https://copy-driven-dev.tistory.com/m/78

[백준][Python][17404][DP] RGB거리 2

풀이법은 첫 번째 조건을 맞추기 위해 25번째 줄과 같이 끝과 처음이 같은 경우를 조건문을 걸어 계산범위에 제외시킨다. 3번 째 조건은 12-16번째 줄에 있는 것으로서 출발 값을 하나로 설정해놓음으로써 i-1번째만 신경쓰면 된다. 2번째 조건은 3번째 조건 안에 들어간다고 보면 된다.

 

나머지 부분은 일반 RGB거리 문제와 같다.