색칠 비용의 최솟값을 구하는 dp문제이다. dp[0]
은 각 색칠 비용의 첫번째 값으로 초기화를 해주고 i=1
부터 반복문을 돌려준다. 반복문을 돌면서 dp[i][j]
는 연속적으로 같은 색을 칠하지 않는 조건에 맞춰 색마다 현재 비용 + 이전 비용 값을 저장해준다. 결과적으로 dp[N-1]
에 3가지 값이 저장되게 되는데 이중 최솟값을 출력해주었다.
어렵지 않게 해결할 수 있었던 dp문제였다.
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int N, res = 987654321;
vector<int> v[3];
int dp[1000][3];
void solution() {
dp[0][0] = v[0][0];
dp[0][1] = v[1][0];
dp[0][2] = v[2][0];
for (int i = 1; i < N; i++) {
for (int j = 0; j < 3; j++) {
if (j == 0) {
dp[i][j] = v[j][i] + min(dp[i - 1][1], dp[i - 1][2]);
}
else if (j == 1) {
dp[i][j] = v[j][i] + min(dp[i - 1][0], dp[i - 1][2]);
}
else if (j == 2) {
dp[i][j] = v[j][i] + min(dp[i - 1][0], dp[i - 1][1]);
}
}
}
for (int i = 0; i < 3; i++) {
res = min(res, dp[N - 1][i]);
}
cout << res;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
cin >> N;
for (int i = 0; i < N; i++) {
int r, g, b;
cin >> r >> g >> b;
v[0].push_back(r);
v[1].push_back(g);
v[2].push_back(b);
}
solution();
return 0;
}