전구

작성일 | In

[백준] 2449번 - 전구

문제 링크

KOI 2011 초등부 문제.
동적 계획법을 이용하여 풀 수 있다.

2차원 dp배열을 사용한다. dp[start][end]는 start~end 구간의 전구 불빛을 같게 만들 때 필요한 최소 색바꿈 횟수이다.
i가 start와 end 사이 인덱스일 때, dp[start][end]는 min(dp[start][end], dp[start][i] + dp[i+1][start] + diff)로 나타낼 수 있다. 여기서 diff는 왼쪽, 오른쪽 구역이 다른 색상인지에 따라 달라진다. 만약 start~i 부분이 색상 1이고, i+1~end가 2라면 둘을 합칠 때 둘 중 하나의 색상으로 통일해야 하므로 diff는 1이 된다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include <bits/stdc++.h>
using namespace std;

#define INF 987654321

int n,k;
vector<int> bulb;
int dp[200][200];

int solve(int start, int end) {
    if(start == end)
        return 0;

    int& ret = dp[start][end];
    if(ret == -1) {
        ret = INF;
        for(int i=start; i<end; ++i) {
            int left = solve(start, i);
            int right = solve(i + 1, end);
            int diff;
            if(bulb[start] == bulb[i+1]) diff = 0;
            else diff = 1;

            ret = min(ret, left + right + diff);
        }
    }
    return ret;
}

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

    bulb.resize(n);
    for(int i=0; i<n; ++i) {
        scanf("%d", &bulb[i]);
    }

    for(int i=0; i<n; ++i)
        fill(dp[i], dp[i] + n, -1);

    printf("%d", solve(0, n-1));
}
0%