这道题是一道反转的开关问题,可以先指定一行的翻转方法,这样每个格子只有它下方的格子能够改变它的颜色。
这道题可以用DFS搜索,也可以用枚举来做。
第一行的状态有2^n次,因此该算法的复杂度为O(mn2^n),反转题的关键在于明白某个格子在被多次翻转后的状态以及检查翻转后剩下的最后一行是否全白。
#include <cstdio>
#include <cstring>
const int maxm = 15;
const int maxn = 15;
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]; //保存中间结果
//查询(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 (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 j = 0; j < n; j++) {
if (get(m-1, j) != 0) {
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;
//按照字典序尝试第一行的所有可能性
for (int i = 0; i < 1 << n; i++) {
memset(flip, 0, sizeof(flip));
for (int j = 0; j < n; j++) {
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(int argc, char const *argv[]) {
scanf("%d%d", &m, &n);
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
scanf("%d", &tile[i][j]);
}
}
solve();
return 0;
}