poj1222 EXTENDED LIGHTS OUT 【高斯消元】

链接：http://poj.org/problem?id=1222

题意：给你一个5*6的矩阵，每个矩阵里有一个灯，用01表示开关状态，每按下一个灯的开关，它与他四周的灯的状态改变，问怎么操作使得灯全都关闭灭。

分析：经典的开关问题，一共有30个灯，用x0~x29表示，每个灯对应一个方程：

x(i*6+j)+x(i*6+j+1)+x(i*6+j-1)+x((i+1)*6+j)+x((i-1)*6+j)%2=a，所以有30个方程。

30个方程，30个未知量，用高斯消元求解。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 1010
#define Mm 2000005
#define mod 1000000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
int a[35][35];
int x[35];
int var,equ;
void gauss() {
    int k,col,max_r;
    for(col=0,k=0;k<equ&&col<var;k++,col++) {
        max_r=k;
        for(int i=k+1;i<equ;i++)
            if(abs(a[i][col])>abs(a[max_r][col])) max_r=i;
        if(max_r!=k)
            for(int j=col;j<=var;j++) swap(a[k][j],a[max_r][j]);
        if(a[k][col]==0) {
            k--;
            continue;
        }
        for(int i=k+1;i<equ;i++) {
            if(a[i][col]) {
                for(int j=col;j<=var;j++) {
                    a[i][j]^=a[k][j];
                }
            }
        }
    }
    for(int i=var-1;i>=0;i--) {
        x[i]=a[i][var];
        for(int j=i+1;j<var;j++) {
            x[i]^=(a[i][j]&&x[j]);
        }
    }
}
void init() {
    CLR(a,0);
    var=30,equ=30;
    for(int i=0;i<5;i++) {
        for(int j=0;j<6;j++) {
            int k=i*6+j;
            a[k][k]=1;
            if(i>0) a[k-6][k]=1;
            if(i<4) a[k+6][k]=1;
            if(j>0) a[k-1][k]=1;
            if(j<5) a[k+1][k]=1;
        }
    }
}
int main() {
    int t;
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++) {
        init();
        for(int i=0;i<30;i++)
            scanf("%d",&a[i][30]);
        gauss();
        printf("PUZZLE #%d",cas);
        for(int i=0;i<30;i++) {
            if(i%6==0) printf("\n%d",x[i]);
            else printf(" %d",x[i]);
        }
        printf("\n");
    }
    return 0;
}