본문 바로가기

백준

1149 RGB거리

 

이것도 디피문제

 

처음에 단순히 각 단계별 최소값을 구하려 했는데 그러면 놓치는 부분이 분명 있을 것 같아서 고민 좀 했다. 

 

알고보니 색을 따로 저장하면 된다. 현재 색이 빨강일 때 -> 이전 값 중 파랑이랑 녹색 중 최소를 더함

빨, 초, 파 전부 구해주면 완성~~~

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()
#define MAXN 1e6

using namespace std;
using i64 = long long;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;

int main() {
    int n;
    scanf("%d", &n);

    vector<vector<int> > v(n + 5, vector<int>(3, 0));

    for (int i = 1; i <= n; i++)
        scanf("%d %d %d", &v[i][0], &v[i][1], &v[i][2]);

    vector<vector<int> > count(n + 5, vector<int>(3, 0));
    for (int i = 1; i <= n; i++)
    {
        count[i][0] = min(count[i - 1][1], count[i - 1][2]) + v[i][0];
        count[i][1] = min(count[i - 1][0], count[i - 1][2]) + v[i][1];
        count[i][2] = min(count[i - 1][1], count[i - 1][0]) + v[i][2];
    }

    printf("%d\n", min({ count[n][0], count[n][1], count[n][2] }));

    return 0;
}

'백준' 카테고리의 다른 글

2565 전깃줄  (0) 2020.08.02
11053 가장 긴 증가하는 부분 수열  (0) 2020.08.02
1463 1로 만들기  (0) 2020.07.31
1003 피보나치 함수  (0) 2020.07.31
4706 쌍둥이 역설  (0) 2020.07.29