[백준] 17404번 : RGB거리 2 (C++)

17404번 : RGB거리 2

---

문제)

RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.

집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.

• 1번 집의 색은 2번, N번 집의 색과 같지 않아야 한다.
• N번 집의 색은 N-1번, 1번 집의 색과 같지 않아야 한다.
• i(2 ≤ i ≤ N-1)번 집의 색은 i-1, i+1번 집의 색과 같지 않아야 한다.

 

 

 

입력 :

첫째 줄에 집의 수 N(2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄부터 N개의 줄에는 각 집을 빨강, 초록, 파랑으로 칠하는 비용이 1번 집부터 한 줄에 하나씩 주어진다. 집을 칠하는 비용은 1,000보다 작거나 같은 자연수이다.

 

 

 

출력 :

첫째 줄에 모든 집을 칠하는 비용의 최솟값을 출력한다.

 

 

 

 

 

풀이)

탑다운 풀이

 

// 17404. RGB거리 2
#include <iostream>
#include <vector>
#include <climits>
 
using namespace std;
 
int n;
vector<vector<int>> vec;
 
int dp[1001][3][3];
int ans = INT_MAX;
 
int recur(int cur, int prv, int first)
{
    if (cur == n) return 0;
 
    int ret = INT_MAX;
 
    if (dp[cur][prv][first] != -1) return dp[cur][prv][first];
 
    for (int i = 0; i < 3; i++)
    {
        if (i == prv) continue;
        if (cur == n - 1 && first == i) continue;
 
        if (cur == 0)
        {
            ret = min(ret, recur(cur + 1, i, i) + vec[cur][i]);
        }
        else
        {
            ret = min(ret, recur(cur + 1, i, first) + vec[cur][i]);
        }
    }
 
    dp[cur][prv][first] = ret;
 
    return dp[cur][prv][first];
}
 
int main()
{
    cin >> n;
    vec.resize(n, vector<int>());
    fill(&dp[0][0][0], &dp[1000][2][3], -1);
 
 
    int a, b, c;
    for (int i = 0; i < n; i++)
    {
        cin >> a >> b >> c;
        vec[i].push_back(a);
        vec[i].push_back(b);
        vec[i].push_back(c);
    }
 
    cout << recur(0, 3, 3);
    
    return 0;
}
 
 

 

바텀업 풀이

 

// 17404. RGB거리 2
#include <iostream>
#include <vector>
#include <climits>
 
using namespace std;
 
int n;
vector<vector<int>> vec;
int dp[1001][3][3];
 
int main()
{
    cin >> n;
    vec.resize(n, vector<int>());
    fill(&dp[0][0][0], &dp[1000][2][3], -1);
 
    int a, b, c;
    for (int i = 0; i < n; i++)
    {
        cin >> a >> b >> c;
        vec[i].push_back(a);
        vec[i].push_back(b);
        vec[i].push_back(c);
    }
 
    for (int i = 0; i < 3; i++)
    {
        for (int j = 0; j < 3; j++)
        {
            dp[n][i][j] = 0;
        }
    }
 
    for (int i = n - 1; i >= 0; i--)
    {
        for (int j = 0; j < 3; j++)
        {
            for (int k = 0; k < 3; k++)
            {
                dp[i][j][k] = INT_MAX;
 
                for (int l = 0; l < 3; l++)
                {
                    if (l == j) continue;
                    if (i == n - 1 && k == l) continue;
 
                    if (i == 0)
                    {
                        dp[i][j][k] = min(dp[i][j][k], dp[i + 1][l][l] + vec[i][l]);
                    }
                    else
                    {
                        dp[i][j][k] = min(dp[i][j][k], dp[i + 1][l][k] + vec[i][l]);
                    }
                }
            }
        }
    }
 
 
    int ans = INT_MAX;
    for (int i = 0; i < 3; i++)
    {
        for (int j = 0; j < 3; j++)
        {
            ans = min(ans, dp[0][i][j]);
        }
    }
 
    cout << ans;
    
    return 0;
}
 
 

 

---

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

17404번: RGB거리 2