알고리즘/백준

RGB거리2 17404번 백준

글을 쓰는 개발자 2020. 12. 4. 19:39
반응형

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

 

17404번: RGB거리 2

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

www.acmicpc.net

 

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

문제 : https://www.acmicpc.net/problem/17404 17404번: RGB거리 2 RGB거리에 사는 사람들은 집을 빨강, 초록, 파랑중에 하나로 칠하려고 한다. 또한, 그들은 모든 이웃은 같은 색으로 칠할 수 없다는 규칙도 정.

copy-driven-dev.tistory.com

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

 

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

반응형