[백준][DP] RGB거리2

1. 문제 주소

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

17404번: RGB거리 2

2. 문제

이전에 풀었던 RGB거리에서 첫 번째 집의 색과 N번째 집의 색이 달라야 한다는 조건이 추가된 문제이다. 이 조건을 추가하여 최솟값을 구하면 된다. 

3. 소스코드

#include <iostream>
#include <algorithm>
using namespace std;
 
int total_price[1001][3];
int color_price[1001][3];
 
int main() {
    int n;
    cin >> n;
    
    for (int i = 1; i <= n; i++) {
        for (int j = 0; j < 3; j++) {
            cin >> color_price[i][j];
        }
    }
    int result = 1000 * 1001;
    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < 3; j++) {
            if (i == j) {
                total_price[1][j] = color_price[1][j];
            }
            else {
                total_price[1][j] = 1000 * 1001;
            }
        }
        for (int k = 2; k <= n; k++) {
            total_price[k][0] = min(total_price[k - 1][1], total_price[k - 1][2]) + color_price[k][0];
            total_price[k][1] = min(total_price[k - 1][0], total_price[k - 1][2]) + color_price[k][1];
            total_price[k][2] = min(total_price[k - 1][0], total_price[k - 1][1]) + color_price[k][2];
        }
        for (int j = 0; j < 3; j++) {
            if(i != j)
            result = min(result, total_price[n][j]);
        }
    }
 
    cout << result << '\n';
    return 0;
}