因为上一行的状态仅与当前行有关
所以我们可以考虑枚举第一行的1<<6个状态
之后再从第2行到第6行开始根据上一行的状态而改变
时间复杂度是O((1<<6)*5*6)
代码如下:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int a[10][10];
int b[10][10];
int c[10][10];
void change(int x, int y) {
a[x][y] = 1-a[x][y];
a[x-1][y] = 1-a[x-1][y];
a[x+1][y] = 1-a[x+1][y];
a[x][y-1] = 1-a[x][y-1];
a[x][y+1] = 1-a[x][y+1];
return ;
}
void dfs(int x) {
for(int i=0; i<6; ++i) {
if(x & (1<<i)) {
b[1][i+1] = 1;
change(1, i+1);
}
}
return ;
}
void print() {
for(int i=1; i<=5; ++i) {
printf("%d", b[i][1]);
for(int j=2; j<=6; ++j)
printf(" %d", b[i][j]);
puts("");
}
return ;
}
bool solve() {
for(int i=2; i<=5; ++i) {
for(int j=1; j<=6; ++j) {
if(a[i-1][j] == 1) {
b[i][j] = 1;
change(i, j);
}
}
}
for(int i=1; i<=5; ++i) {
for(int j=1; j<=6; ++j)
if(a[i][j])
return false;
}
return true;
}
int main(void) {
int n;
scanf("%d", &n);
for(int _=1; _<=n; ++_) {
memset(a, 0, sizeof(a));
for(int i=1; i<=5; ++i)
for(int j=1; j<=6; ++j)
scanf("%d", &c[i][j]);
for(int i=0; i<(1<<6); ++i) {
memcpy(a, c, sizeof(c));
memset(b, 0, sizeof(b));
dfs(i);
/*
puts("");
puts("");
puts("");
*/
if(solve()) {
// printf("b--->\n");
printf("PUZZLE #%d\n", _);
print();
break;
}
/*
printf("c--->\n");
for(int i=1; i<=5; ++i) {
for(int j=1; j<=6; ++j) {
printf("%d", c[i][j]);
}
puts("");
}
puts("");
puts("");
puts("");
*/
}
}
return 0;
}
高斯消元法的做法是列出30个方程,并由这些方程求出30个未知数
而这里构造方程的方法很值得学习以下
因为一个方程中可能含3、4、5个未知数
对应一个点周围的点
由此可以构造一个30×30的矩阵
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int a[100][100];
void gauss() {
int i, j, k, r;
for(i=0; i<30; ++i) {
r = i;
for(j=i+1; j<30; ++j)
if(a[j][i])
r = j;
if(r != i) {
for(k=0; k<=30; ++k)
swap(a[i][k], a[r][k]);
}
for(k=0; k<30; ++k) {
if(k!=i && a[k][i])
for(j=0; j<=30; ++j)
a[k][j] ^= a[i][j];
}
}
return ;
}
int main(void) {
int n;
scanf("%d", &n);
for(int _=1; _<=n; ++_) {
for(int i=0; i<30; ++i) {
scanf("%d", &a[i][30]);
}
for(int i=0; i<30; ++i) {
a[i][i] = 1;
if(i%6 != 0) a[i-1][i] = 1;
if(i%6 != 5) a[i+1][i] = 1;
if(i > 5) a[i-6][i] = 1;
if(i < 24) a[i+6][i] = 1;
}
gauss();
printf("PUZZLE #%d\n", _);
for(int i=0; i<30; ++i) {
printf("%d", a[i][30]);
if(i%6 == 5) puts("");
else printf(" ");
}
}
return 0;
}