POJ 1222 EXTENDED LIGHTS OUT [高斯消元]

　　为了学习高斯消元写了下这一题，代码是按照维基百科的伪代码来写的，每个点的状态和他自己以及周围五个点是否按下有关，于是只要每个点作为一个未知数，列出30个方程，用高斯消元求解即可。

　　其实这题完全可以不必用高斯消元，枚举第一行的状态就可以了，第一行的状态可以决定第二行的状态，进而就可以决定整幅图的状态。

　　

#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
int cas,n,m,x[50][50],ans[31];
void gauss(){
    int i=1,j=1;
    while(i<=n&&j<=m){
        int maxi=i;
        for(int k=i+1;k<=n;k++)
            if(abs(x[k][j])>abs(x[maxi][j]))maxi=k;
        if(x[maxi][j]){
            for(int k=1;k<=m;k++)std::swap(x[i][k],x[maxi][k]);
            for(int k=j,div=x[i][j];k<=m;k++)x[i][k]/=div;
            for(int k1=i+1;k1<=n;k1++)
             for(int k2=j,y=x[k1][j];k2<=m;k2++)
              x[k1][k2]-=y*x[i][k2],x[k1][k2]=x[k1][k2]%2;
            i++;
        }
        j++;
    }
    ans[31]=1;
    for(int i=n;i>=1;i--){
        int tmp=0;
        for(int j=i+1;j<=31;j++)
            tmp+=ans[j]*x[i][j];
        ans[i]=(tmp%2+2)%2;
    }
}

int main(){
   //freopen("test.in","r",stdin);
    scanf("%d",&cas);
    for(int ca=1;ca<=cas;ca++){
        memset(x,0,sizeof x);
        n=30,m=31;
        for(int i=1;i<=30;i++){
            x[i][i]=x[i][i-6]=x[i][i+6]=1;
            if(i%6!=1)x[i][i-1]=1;
            if(i%6!=0)x[i][i+1]=1;
        }
        for(int i=1;i<=30;i++)scanf("%d",&x[i][31]);
        gauss();
        printf("PUZZLE #%d\n",ca);
        for(int i=1;i<=30;i++){
            printf("%d",ans[i]);
            printf(i%6?" ":"\n");
        }
    }
    return 0;
}