본문 바로가기
Algorithm/동적계획법(DP)

1149 RGB 거리 (복기)

by neohtux 2020. 3. 20.
728x90

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

 

1149번: RGB거리

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

www.acmicpc.net

RGB거리 성공

시간 제한메모리 제한제출정답맞은 사람정답 비율

0.5 초 (추가 시간 없음) 128 MB 41405 19303 14510 47.572%

문제

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. 점화식을 찾아본다.  D[N][x색] = N개의 집을 x색으로 칠했을 경우의 수

 

2. 마지막 집을 기준으로

R로 끝났는지

G로 끝났는지

B로 끝났는지에 대해

각각 나뉘어 점화식을 정리 예를 들어 R로 칠했을경우 D[N][R] = min(D[N-1][G] + G색값 , D[N-1][B] + B색값);

 

3. 나머지는 구현

 

4. 한달이 지나 복기해보니 이런 유형의 점화식이 이제 보인다...

 

#include<iostream>
#include<algorithm>
using namespace std;


int d[1001][3]; //rgb : 012
int color[1001][3];
int main(void)
{
	int N;
	cin >> N;

	for (int i = 1; i <= N; ++i)
	{
		cin >> color[i][0] >> color[i][1] >> color[i][2];
	}

	d[0][0] = 0;
	d[0][1] = 0;
	d[0][2] = 0;
	for (int i = 1; i <= N; ++i)
	{
		d[i][0] = min(d[i - 1][1] + color[i][1], d[i - 1][2] + color[i][2]);
		d[i][1] = min(d[i - 1][0] + color[i][0], d[i - 1][2] + color[i][2]);
		d[i][2] = min(d[i - 1][0] + color[i][0], d[i - 1][1] + color[i][1]);
	}

	int ans = d[N][0];
	if (ans > d[N][1]) ans = d[N][1];
	if (ans > d[N][2]) ans = d[N][2];

	cout << ans << '\n';
	return 0;
}
300x250

'Algorithm > 동적계획법(DP)' 카테고리의 다른 글

11057 오르막 수  (0) 2020.03.24
1309 동물원  (0) 2020.03.24
1699 제곱수의 합  (0) 2020.03.19
1912 연속합  (0) 2020.03.19
15990 1,2,3 더하기 5  (0) 2020.03.19

댓글