poj 1222 EXTENDED LIGHTS OUT

http://poj.org/problem?id=1222

高斯消元

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<cstring>

#include<map>

#include<cmath>

#define LL long long



using namespace std;



const int N=35;



int lx[4]={1,-1,0,0};

int ly[4]={0,0,1,-1};

int a[N][N];

int ans[N];

int n;

void Gauss()

{

    int row=0;

    int col=0;

    while(row<n&&col<n)

    {

        int k=-1;

        for(int i=row;i<n;++i)

        {

            if(a[i][col]!=0)

            {

                k=i;break;

            }

        }

        if(k!=row)

        {

            for(int j=col;j<=n;++j)

            {

                swap(a[row][j],a[k][j]);

            }

        }

        int x=a[row][col];

        for(int i=row+1;i<n;++i)

        {

            int y=a[i][col];

            for(int j=col;j<=n;++j)

            {

                a[i][j]=(a[i][j]*x-y*a[row][j])%2;

            }

        }

        ++row;++col;

    }

    //cout<<row<<" "<<col<<endl;

    for(int i=row-1;i>=0;--i)

    {

        int sum=0;

        for(int j=i+1;j<n;++j)

        sum=sum+(ans[j]*a[i][j]);

        ans[i]=(a[i][n]-sum)/a[i][i]%2;

    }

    for(int i=0;i<5;++i)

    {

        for(int j=0;j<6;++j)

        {

            printf("%d",(ans[i*6+j]+2)%2);

            if(j<5)

            printf(" ");

        }

        printf("\n");

    }

}

int main()

{

    n=30;

    int T;

    scanf("%d",&T);

    for(int w=1;w<=T;++w)

    {

        memset(a,0,sizeof(a));

        for(int i=0;i<5;++i)

        {

            for(int j=0;j<6;++j)

            {

                int x;

                scanf("%d",&x);

                if(x==1)

                a[i*6+j][n]=1;

                a[i*6+j][i*6+j]=1;

                for(int l=0;l<4;++l)

                {

                    int l1=i+lx[l];

                    int l2=j+ly[l];

                    if(l1>=0&&l1<5&&l2>=0&&l2<6)

                    {

                        a[l1*6+l2][i*6+j]=1;

                    }

                }

            }

        }

        printf("PUZZLE #%d\n",w);

        Gauss();

    }

    return 0;

}