백준 1149번 - RGB거리 (C++)

해결 방법

DP문제.
n*3 크기의 dp에 현재 수와, 그 이전까지 의 합(dp)를 더해서 최대값을 찾아 저장한다.
이후 가장 마지막 줄의 3칸중 가장 낮은 값이 정답.

코드

#include <iostream>
using namespace std;
 
int dp[3][1001];
int rgb[3][1001];
 
int GetMin(int a, int b)
{
    if(a==0)
        return b;
    if(b==0)
        return a;
    
    return a > b ? b : a;
}
 
int main(int argc, char* argv[])
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    int n;
    cin>>n;
 
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<3;j++)
            cin>>rgb[j][i];
    }
 
    dp[0][0] = rgb[0][0];
    dp[1][0] = rgb[1][0];
    dp[2][0] = rgb[2][0];
 
    for(int i=0;i<n-1;i++)
    {
        for(int j=0;j<3;j++)
        {
            if (j == 0)
            {
                dp[1][i+1] = GetMin(dp[1][i+1], dp[j][i] + rgb[1][i + 1]);
                dp[2][i+1] = GetMin(dp[2][i+1], dp[j][i] + rgb[2][i + 1]);
            }
            else if (j == 1)
            {
                dp[0][i+1] = GetMin(dp[0][i+1], dp[j][i] + rgb[0][i + 1]);
                dp[2][i+1] = GetMin(dp[2][i+1], dp[j][i] + rgb[2][i + 1]);
            }
            else
            {
                dp[0][i+1] = GetMin(dp[0][i+1], dp[j][i] + rgb[0][i + 1]);
                dp[1][i+1] = GetMin(dp[1][i+1], dp[j][i] + rgb[1][i + 1]);
            }
        }
    }
    
    cout << GetMin(GetMin(dp[0][n-1],dp[1][n-1]), dp[2][n-1]);;
    
    return 0;
}