BOJ)14939 불 끄기

문제: icpc.me/14939 

언제 봤는지는 기억이 잘 안 나지만 언젠가 한번 봤었던 은근히 well-known 문제이다.

첫 줄에 전구를 어떤 방식으로 뒤집는 걸 결정을 했다고 가정하면 그 아래 줄은 바로 위의 전구가 켜져 있는지 여부에 따라서 뒤집기 결정을 그리디하게 할 수 있다.

따라서 첫 줄을 뒤집는 모든 경우 (2^10)에 대하여 정해준 뒤, 그리디하게 뒤집어주어 최적의 답을 구하는 것이다. 

비슷한 시기에 나온 홍익대학교 경시대회의 전구 끄기 문제도 한번 풀어보면 좋을 것 같다.

#include <cstdio>
#include <algorithm>
using namespace std;
int n, a[11][11], b[11][11], res;
char in[11][11];
int dx[] = { 0,0,1,-1 };
int dy[] = { 1,-1,0,0 };
int chk(int x, int y) {
    return 0 <= x&&x < n && 0 <= y&&y < n;
}
void swp(int x, int y) {
    b[x][y] ^= 1;
    for (int i = 0; i < 4; i++) {
        int cx = x + dx[i];
        int cy = y + dy[i];
        if (!chk(cx, cy))continue;
        b[cx][cy] ^= 1;
    }
}
int main() {
    res = 1e9, n = 10;
    for (int i = 0; i < n; i++) {
        scanf("%s", &in[i]);
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++)
            if (in[i][j] == 'O')a[i][j] = 1;
    }
    for (int i = 0; i < (1 << n); i++) {
        for (int j = 0; j < n; j++) {
            for (int k = 0; k < n; k++)
                b[j][k] = a[j][k];
        }
        int cnt = 0;
        for (int j = 0; j < n; j++) {
            if ((1 << j)&i) {
                cnt++;
                swp(0, j);
            }
        }
        for (int j = 1; j < n; j++) {
            for (int k = 0; k < n; k++) {
                if (b[j - 1][k]) {
                    cnt++;
                    swp(j, k);
                }
            }
        }
        int f = 0;
        for (int j = 0; j < n; j++) {
            for (int k = 0; k < n; k++)
                f += b[j][k];
        }
        if (!f)
            res = min(res, cnt);
    }
    printf("%d\n", res == 1e9 ? -1 : res);
    return 0;
}