Fliptile POJ - 3279

搜索

先枚举确定第一排，之后的都可以确定

#include<cstdio>
#include<cstring>

using namespace std;

const int MAXN = 16;
const int MAXM = 16;

//邻接的格子的坐标
const int dx[5] = {-1,0,0,0,1};
const int dy[5] = {0,1,0,-1,0};

//输入
int M,N;
int tile[MAXM][MAXN];
//保存最优解
int opt[MAXM][MAXN];
//保存中间结果
int flip[MAXM][MAXN];

//find the color of point(x,y)
int get(int x, int y) {
    //原始颜色
    int c = tile[x][y];
    for(int i = 0; i < 5; ++i) {
        //被相邻的格子影响过之后的结果
        int x2 = x + dx[i], y2 = y + dy[i];
        if(0 <= x2 && x2 < M && 0 <= y2 && y2 < N) {
            c += flip[x2][y2];
        }
    }
    return c % 2;
}

//求出第一行确定情况下的最小操作次数
//不存在解返回-1
int calc() {
    //求出从第二行开始的反转方法
    for(int i = 1; i < M; i++) {
        for(int j = 0 ; j < N; j++) {
            if(get(i-1,j)!= 0) {
                // (i-1,j)是黑色的话，就必须反转这个格子
                flip[i][j] = 1;
            }
        }
    }
    //判断最后一行是否全白
    for(int i = 0; i < N; ++i) {
        if(get(M-1,i) !=0) {
            // no solution
            return -1;
        }
    }

    // 统计反转次数
    int res = 0;
    for(int i = 0 ; i < M; ++i) {
        for(int j = 0; j < N; ++j) {
            res += flip[i][j];
        }
    }
    return res;
}

int slove() {
    int res = -1;
    for(int i = 0; i < 1 << N; ++i) {
        //按照字典序尝试第一行,
        memset(flip, 0, sizeof(flip));
        for(int j = 0; j < N; ++j) {
            // 因为是按照字典序，所以后面先来翻后面的,就可以保证结果字典序小的在前面
            // 把i当作一串2进制数，将此二进制数的倒数第k位（与1即可取出）赋值给flip[0][k]
            flip[0][N-j-1] = i >> j & 1;
        }
        //获取反转结果
        int num = calc();
        //更新结果，将此次的复制到答案
        if(num >=0 &&(res<0||res>num)) {
            res = num;
            memcpy(opt, flip, sizeof(flip));
        }
    }
    if(res < 0) {
        printf("IMPOSSIBLE\n");
    } else {
        for(int i = 0; i < M; ++i) {
            for(int j = 0; j < N; ++j) {
                printf("%d%c", opt[i][j], j + 1 == N ? '\n': ' ');
            }
        }
    }
}

int main() {
    while(scanf("%d%d",&M,&N) == 2) {
        for(int i = 0; i < M; ++i) {
            for(int j = 0; j < N; ++j) {
                scanf("%d", &tile[i][j]);
            }
        }
        slove();
    }
    return 0;
}