poj1222 高斯消元

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':' ');
    }
}