poj1222 高斯消元

#include <iostream>
#include <algorithm>
#include <string.h>
#include <cstdio>
#include <vector>
using namespace std;

#define eps 1e-9
const int MAXN=35;
int a[MAXN][MAXN],x[MAXN];//方程的左边的矩阵和等式右边的值，求解之后x存的就是结果
int equ,var;//方程数和未知数个数
int Gauss()
{
    int i,j,row,col,max_r;
    equ=var=30;
    for(row=0,col=0;row<equ&&col<var;row++,col++)
    {
        max_r=row;
        for(i=row+1;i<equ;i++)
          if(a[i][col])
            { max_r=i; break; }
        if(row!=max_r)
        {

            for(j=col;j<var;j++)
              swap(a[row][j],a[max_r][j]);
            swap(x[row],x[max_r]);

        }
        for(i=0;i<equ;i++)
          if(i!=row&&a[i][col])
          {
              x[i]^=x[row];
              for(j=col+1;j<var;j++)a[i][j]=a[i][j]^a[row][j];
              a[i][col]=0;
          }
    }
    return 1;
}
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};
int idmark[35][35];
int main()
{
    for(int i=0; i<5; i++)
        for(int j=0; j<6; j++)
           idmark[i][j]=i*6+j;
    int cas;
    scanf("%d",&cas);
    for(int cc=1; cc<=cas ; cc++)
    {
       for(int i=0; i<30; i++)
        {
            scanf("%d",&x[i]);
        }
       memset(a,0,sizeof(a));
       for(int i=0; i<5; i++)
       for(int j=0; j<6; j++){
              a[ idmark[i][j] ][ idmark[i][j] ] = 1;

          for(int k=0; k<4; k++){

              int tx= i+dx[k];

              int ty= j+dy[k];

              if(tx>=0&&ty>=0&&tx<5&&ty<6)
              a[ idmark[i][j] ][ idmark[tx][ty] ]=1;

          }
       }
       Gauss();
       printf("PUZZLE #%d\n",cc);
       int cur=0;
       for(int i=0; i<5; i++)
       for(int j=0; j<6; j++)
        printf("%d%c",x[cur++],j==5?‘\n‘:‘ ‘);
    }


    return 0;
}