백준 1149_RGB거리 / Python
2024. 2. 4. 23:35ㆍ백준
문제
RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.
집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.
- 1번 집의 색은 2번 집의 색과 같지 않아야 한다.
- N번 집의 색은 N-1번 집의 색과 같지 않아야 한다.
- i(2 ≤ i ≤ N-1)번 집의 색은 i-1번, i+1번 집의 색과 같지 않아야 한다.
입력
첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나 같은 자연수이다.
출력
첫째 줄에 모든 집을 칠하는 비용의 최솟값을 출력한다.
예제 입력 1
3
26 40 83
49 60 57
13 89 99
예제 출력 1
96
예제 입력 2
3
1 100 100
100 1 100
100 100 1
예제 출력 2
3
예제 입력 3
3
1 100 100
100 100 100
1 100 100
예제 출력 3
102
예제 입력 4
6
30 19 5
64 77 64
15 19 97
4 71 57
90 86 84
93 32 91
예제 출력 4
208
예제 입력 5
8
71 39 44
32 83 55
51 37 63
89 29 100
83 58 11
65 13 15
47 25 29
60 66 19
예제 출력 5
253
## 1149 RGB거리
n=int(input())
lst=[]
for _ in range(n):
lst.append(list(map(int,input().split(' '))))
def min_rgb(lst):
for i in range(1,n): #0번째집 비용은 이미 주어져있으므로 계산할 필요X
lst[i][0]+=min(lst[i-1][1],lst[i-1][2]) #현재 집을 빨강으로 칠함, 이전 집을 초록/파랑으로 칠하는 것 중 작은 값.
lst[i][1]+=min(lst[i-1][0],lst[i-1][2]) #현재 집을 초록으로 칠함, 이전 집을 빨강/파랑으로 칠하는 것 중 작은 값.
lst[i][2]+=min(lst[i-1][0],lst[i-1][1]) #현재 집을 파랑으로 칠함, 이전 집을 빨강/초록으로 칠하는 것 중 작은 값.
return min(lst[-1]) #최종적인 최소 비용 return
print(min_rgb(lst))
다음 집을 어떻게 칠할지로 접근하는 것 보다 이전 집을 뭘로 칠했는지로 접근하는게 훨씬 쉬웠다.
'백준' 카테고리의 다른 글
백준 9461_파도반 수열 / Python (0) | 2024.02.06 |
---|---|
백준 2579_계단 오르기 / Python (1) | 2024.02.06 |
백준 14888_연산자 끼워넣기 / Python (2) | 2024.01.23 |
백준 10819_차이를 최대로 / Python (1) | 2024.01.23 |
백준 14501_퇴사 / Python (0) | 2024.01.22 |