BOJ 1149번 RGB거리

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

github : https://github.com/junho0956/Algorithm/blob/master/1149/1149/%EC%86%8C%EC%8A%A4.cpp

 

재귀를 통해서 RGB거리에 대한 최소합을 구현하였습니다.

범위를 벗어나면 dp의 최소값을 계속 적용시켜주기 위해 maxi를 반환하였고

범위내에 포함만 된다면 dp값을 계속 확인해가면서 칠할 수 있는 최소값을 저장해두었습니다.

 

#include <iostream>
#include <algorithm>
using namespace std;
 
#define maxi 987654321
int dp[1000][3];
int arr[1000][3];
int n;
 
int dfs(int house, int color) {
    if (color < 0 || color > 2) return maxi;
    if (house == n - 1) return arr[house][color];
 
    int& res = dp[house][color];
    if (res) return res;
 
    res = maxi+10000;
 
    for (int i = 0; i < 3; i++) {
        if (i != color) res = min(res, dfs(house + 1, i));
    }
    res += arr[house][color];
 
    return res;
}
 
int main() {
    ios::sync_with_stdio(0), cin.tie(0);
 
    cin >> n;
 
    for (int i = 0; i < n; i++)
        for (int j = 0; j < 3; j++)
            cin >> arr[i][j];
 
    int ans = 987654321;
    for (int i = 0; i < 3; i++) {
        ans = min(ans, dfs(0, i));
    }
 
    cout << ans;
    return 0;
}