백준 1149 RGB거리 (C++)

안유태·2022년 9월 6일
0

알고리즘

목록 보기
33/239

1149번: RGB거리

색칠 비용의 최솟값을 구하는 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;
}
profile
공부하는 개발자

0개의 댓글