POJ1222熄灯问题

千年老题，以前用枚举做，现在用高斯消元做

自由元直接做成0即可

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
using namespace std;
int a[55][55];
int ans[6][7];
int c(int x,int y){return (x*6+y+1);}
int d[5]={0,-1,0,1,0};
void gauss(int m,int n){
    int line=1;
    for(int k=1;k<=m;k++){
        int i=line;
        while(i<=m){if(a[i][k])break;i++;}
        if(i>m)continue;
        if(i!=line){swap(a[i],a[line]);}
        for(i=1;i<=m;i++){
            if(i!=line&&a[i][k]){
                for(int j=k;j<=n;j++){
                    a[i][j]^=a[line][j];
                }
            }
        }
        line++;
    }
//    for(int i=1;i<=30;i++){
//        for(int j=1;j<=30;j++){
//            printf("%d ",a[i][j]);    
//        }
//        printf("\n");
//    }
    for(int i=0;i<5;i++){
        for(int j=0;j<6;j++){
            ans[i][j]=a[c(i,j)][n];
        }
    }
}
void solve(){
    memset(a,0,sizeof(a));
    memset(ans,0,sizeof(ans));
    int x;
    for(int i=0;i<5;i++){
        for(int j=0;j<6;j++){
            scanf("%d",&x);
            a[c(i,j)][31]=x;
            a[c(i,j)][c(i,j)]=1;
            for(int k=0;k<4;k++){
                int x=i+d[k],y=j+d[k+1];
                if(x>=0&&x<5&&y>=0&&y<6)a[c(i,j)][c(x,y)]=1;
            }
        }
    }
    gauss(30,31);
    for(int i=0;i<5;i++){
        for(int j=0;j<5;j++){
            printf("%d ",ans[i][j]);
        }
        printf("%d\n",ans[i][5]);
    }
}
int main()
{
//    freopen("data.in","r",stdin); 
    int T;
    scanf("%d",&T);
    for(int i=1;i<=T;i++){
        printf("PUZZLE #%d\n",i);
        solve();
    }
    return 0;
}