poj3279 Fliptile

#include <iostream>
#include <cstring>
using namespace std;
// 邻接格子的坐标
const int dx[5] = {-1, 0, 0, 0, 1};
const int dy[5] = {0, -1, 0, 1, 0};
const int MAX_M = 16;
const int MAX_N = 16;
int M, N;
int tile[MAX_M][MAX_N];

int opt[MAX_M][MAX_N]; // 保存最优解
int flip [MAX_M][MAX_N]; // 保存中间结果

// 查询(x, y)的颜色
int get(int x, int y)
{
int c = tile[x][y];
for (int d = 0; d < 5; d++)
{
int x2 = x + dx[d], y2 = y + dy[d];
if (x2 >= 0 && x2 < M && y2 >= 0 && y2 < N)
{
c += flip[x2][y2];
}
}
return c % 2;
}

// 求出第1行确定情况下的最小操作次数
int calc()
{
// 求出从第2行开始的翻转方法
for (int i = 1; i < M; i++)
for (int j = 0; j < N; j++)
{
if (get(i - 1, j) != 0)
{
// 如果同列的上一行为黑色，则必须翻转，因为此后，只有该点的翻转可以改变正上方的棋子颜色了
flip[i][j] = 1;
}
}

// 判断最后一行是否空白
for (int j = 0; j < N; j++)
if (get(M - 1, j)) // 无解
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;
}

void solve()
{
int res = -1;
memset(opt, 0, sizeof(opt));
// 按照字典序开始尝试第一行所有的可能性
for (int i = 0; i < 1 << N; i++)
{
memset(flip, 0, sizeof(flip)); // 每次必须重新清空flip
for (int j = 0; j < N; j++)
{
flip[0][N - 1 - j] = i >> j & 1;
}

int num = calc();

if (num >= 0 && (res == -1 || res > num))
{
res = num;
memcpy(opt, flip, sizeof(flip));
}
}

if (res == -1) cout << "IMPOSSIBLE" << endl;
else
{
for (int i = 0; i < M; i++)
{
for (int j = 0; j < N - 1; j++)
cout << opt[i][j] << " ";

cout << opt[i][N - 1] << endl;
}
}
}

int main()
{
while (cin >> M >> N)
{
for (int i = 0; i < M; i++)
for (int j = 0; j < N; j++)
cin >> tile[i][j];

solve();
}

return 0;
}