백준 17404 RGB거리 2 파이썬

dasd412·2022년 5월 30일
0

코딩 테스트

목록 보기
43/71

문제 설명

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

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

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

입력 조건

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

출력 조건

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


전체 코드

import sys

n = int(sys.stdin.readline().rstrip())

arr = []

for i in range(n):
    arr.append(list(map(int, sys.stdin.readline().split())))

answer = sys.maxsize

# 아예 첫 번째 집의 색깔을 미리 선택했다고 가정하고 푼다.
for initial_color in range(3):
    # 현재 최솟값을 기록하는 메모이제이션
    dp = [[sys.maxsize] * n for _ in range(3)]
    # 첫 번째 집 미리 색칠해보기
    dp[initial_color][0] = arr[0][initial_color]

    for col in range(1, n):
        # 예를 들어, 이번 col에 빨간 색을 칠한다면, 이전에 파란 색 또는 초록 색 칠한 비용 중 더 작은 값을 택한 후 현재 col의 빨간 색 비용을 더해야 한다.
        dp[0][col] = min(dp[1][col - 1], dp[2][col - 1]) + arr[col][0]
        dp[1][col] = min(dp[0][col - 1], dp[2][col - 1]) + arr[col][1]
        dp[2][col] = min(dp[0][col - 1], dp[1][col - 1]) + arr[col][2]

    # 처음에 선택했던 값과 달라야 한다.
    for row in range(3):
        if row != initial_color:
            answer = min(answer, dp[row][n - 1])

print(answer)

환형 조건이 추가됬을 뿐인데, 기존 RBG 거리보다 문제가 어려웠다...

profile
아키텍쳐 설계와 테스트 코드에 관심이 많음.

0개의 댓글