POJ-1222-开关灯问题-高斯消元

题目：一个5*6的开关矩阵，按一个按钮能使这个灯和上下左右的灯改变状态，问怎么按可以使灯全灭

思路：建立一个方程，用高斯消元求

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#define maxn 30
using namespace std;
int a[50][50];
int loc(int x,int y) {
    return x*6-6+y;
}
void swapl(int n,int m) {
    int tt;
    for(int i=1;i<=31;i++) {
        tt=a[n][i];
        a[n][i]=a[m][i];
        a[m][i]=tt;
    }
    return;
}
void Guass(int n) {
    for(int i=1;i<=n;i++) {
        int k=0;
        for(int j=i;j<=n;j++) {
            if(a[j][i]==1) {
                k=j;
                break;
            }
        }
        if(k!=i) {
            swapl(k,i);
        }
        for(int j=i+1;j<=n;j++) {
            if(a[j][i]==1)
            for(int u=i;u<=n+1;u++)
                a[j][u]^=a[i][u];
        }
    }
    for(int i=30;i>=2;i--) {
        for(int j=i-1;j>=1;j--) {
            if(a[j][i]==1)
            for(int k=i;k<=n+1;k++)
                a[j][k]^=a[i][k];
        }
    }
}
void prit(void) {
    for(int i=1;i<=30;i++) {
        for(int j=1;j<=31;j++)
            cout << a[i][j] << " ";
        cout << endl;
    }
}
int main()
{
    int t;
    int kase=0;
    int xx;
    scanf("%d",&t);
    while(t--){
        for(int i=1;i<=5;i++)
            for(int j=1;j<=6;j++) {
                scanf("%d",&xx);
                a[loc(i,j)][31]=xx;
                a[loc(i,j)][loc(i,j)]=1;
                if(i>1) a[loc(i,j)][loc(i-1,j)]=1;
                if(j>1) a[loc(i,j)][loc(i,j-1)]=1;
                if(i<5) a[loc(i,j)][loc(i+1,j)]=1;
                if(j<6) a[loc(i,j)][loc(i,j+1)]=1;
            }
      //  prit();
        Guass(maxn);
      //  cout << endl;
       // prit();
        printf("PUZZLE #%d\n",++kase);
        for(int i=1;i<=5;i++) {
            for(int j=1;j<=5;j++)
                printf("%d ",a[loc(i,j)][31]);
            printf("%d\n",a[(loc(i,6))][31]);
        }
    }
    return 0;
}