C++ 백준 1149. RGB거리

0

C++ 코딩테스트

목록 보기
10/30

문제

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

풀이

이 문제는 i번째 집을 j의 색깔로 칠할 경우의 값을 저장하고, 차근차근 dp를 사용하여 퓰면 된다.
각 house에 대해 RGB 값으로 색칠을 하는 값을 받고,
i번째 집을 i-1번째 집과 다른 색으로 칠할 때의 결과값을 저장하면 된다.

#include<bits/stdc++.h>
using namespace std;
int n;
int house[1005][3];
int dk[1005][3];
int main(){
	cin >> n;
	for(int i = 0; i < n; i++) cin >> house[i][0] >> house[i][1] >> house[i][2];
	dk[0][0] = house[0][0]; // 초기값 지정
	dk[0][1] = house[0][1];
	dk[0][2] = house[0][2];
	for(int i = 1; i < n; i++){
		dk[i][0] = min(dk[i-1][1],dk[i-1][2]) + house[i][0]; // i번째 집을 'R'로 칠할 때, (i-1)번째 집까지 칠하는 경우의 수 중에 i번째 집의 색깔과 다른 경우의 값을 더해 간다.
		dk[i][1] = min(dk[i-1][0],dk[i-1][2]) + house[i][1]; // i번째 집을 'G'로 칠하는 경우
		dk[i][2] = min(dk[i-1][0],dk[i-1][1]) + house[i][2]; // i번째 집을 'B'로 칠하는 경우
	}
	int ret;
	ret = min({dk[n-1][0],dk[n-1][1],dk[n-1][2]}); // i번째까지 집을 칠할 때의 최솟값 
	cout << ret;
	return 0;
}

0개의 댓글