A[i][j]为1代表第j个开关控制第i盏灯。 那么若第j个灯关着，对应的开关按的次数和为偶，最终为关。 若第j个灯开着，对应的开关按的次数和为奇，最终为关。 构造矩阵高斯消元即可。
#include <cstdio> #include <bitset> #include <cmath> #include <iostream> using namespace std; int x[32]; bitset<32> A[32]; void gao() { int i,k; for(k=0;k<30;k++) { i=k; for(;i<30;i++) if(A[i][k]) break; swap(A[k],A[i]); for(i=0;i<30;i++) if(k!=i&&A[i][k]) A[i]^=A[k]; } for(i=0;i<30;i++) x[i]=A[i][30]; } int main() { int t; scanf("%d",&t); for(int cas=1;cas<=t;cas++) { for(int i=0;i<32;i++) A[i].reset(); for(int i=0;i<5;i++) for(int j=0;j<6;j++) { int x;scanf("%d",&x); if(x) A[i*6+j][30]=1; } for(int i=0;i<5;i++) for(int j=0;j<6;j++) { A[i*6+j][i*6+j]=1; if(i-1>=0) A[(i-1)*6+j][i*6+j]=1; if(j-1>=0) A[i*6+j-1][i*6+j]=1; if(i+1<5) A[(i+1)*6+j][i*6+j]=1; if(j+1<6) A[i*6+j+1][i*6+j]=1; } gao();int cnt=0; printf("PUZZLE #%d\n",cas); for(int i=0;i<5;i++) for(int j=0;j<6;j++) printf("%d%c",x[cnt++],j==5?'\n':' '); } }