[백준] #1149번 Dynamic Programming python

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

1149번: RGB거리

 

DP문제로 분류 되며, 거리 구하기 문제이다.

 

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

 

 

코드는 아래와 같다.

 

 

def main():
    n = int(input())
    dp = []
    for _ in range(n):
        dp.append(list(map(int, input().split())))

    # print(dp)
    for i in range(1, len(dp)):
        dp[i][0] = min(dp[i-1][1], dp[i-1][2]) + dp[i][0]
        dp[i][1] = min(dp[i-1][0], dp[i-1][2]) + dp[i][1]
        dp[i][2] = min(dp[i-1][0], dp[i-1][1]) + dp[i][2]

    print(min(dp[n - 1][0], dp[n - 1][1], dp[n - 1][2]))

if __name__:
    main()

'''
3
26 40 83
49 60 57
13 89 99
'''