728x90
https://www.acmicpc.net/problem/1149
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 |
댓글