POJ 1222 EXTENDED LIGHTS OUT 开关问题

题目链接： 点我
题目大意： 一个5*6的网格，每次反转一个方格及相邻的四个，要把所有的反转成0
题目分析： 暴力枚举第一行的所有反转方案，然后就可以确定剩下4行的方案了。据说用高斯消元也可以，但是蒟蒻的我并不会。（省选的时候就是这题。。。当时用贪心做的，每次只考虑2*2的方格，水了30分。。。）
PS：测试输出没有删干净。。。初始化也没弄清楚，导致全是0的点过不去。。。WA了半多小时

Problem: 1222       User: ChenyangDu
Memory: 156K        Time: 0MS
Language: C++       Result: Accepted
Source Code
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int INF = 0x7fffffff,map[2][5] = {1,0,0,-1,0,0,1,-1,0,0};
int in[10][10],ans[10][10],out[10][10];//ans是中间变量，out是最终结果

int get_color(int x,int y){
    int color = in[x][y];
    for(int i=0;i<5;i++){
        int a = x + map[0][i],
            b = y + map[1][i];
        if(a>=0 && a<5 && b>=0 && b<6 && ans[a][b] == 1){
            color ++;
        }
    }
    return color % 2;
}

int cal(){
    int res = 0;
    for(int i=1;i<5;i++){
        for(int j=0;j<6;j++){
            if(get_color(i-1,j) == 1){
                ans[i][j] = 1;
                res++;
            }
        }
    }
    for(int i=0;i<6;i++){
        if(get_color(4,i))return -1;
    }

    return res;
}

int main(){
    //freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    int t = 0;
    while(t++<T){
        memset(out,0,sizeof(out));
        for(int i=0;i<5;i++){
            for(int j=0;j<6;j++){
                scanf("%d",&in[i][j]);
            }
        }
        memset(ans,0,sizeof(ans));

        for(int i=0;i<(1<<7);i++){//暴力枚举第一行
            memset(ans,0,sizeof(ans));
            for(int j=0;j<6;j++){
                ans[0][6-j-1] = (i>>j)&1;
            }
            if(cal()>0){
                memcpy(out,ans,sizeof(out));
                break;
            }
        }
        printf("PUZZLE #%d\n",t);
        for(int i=0;i<5;i++){
            for(int j=0;j<5;j++)
                printf("%d ",out[i][j]);

            printf("%d\n",out[i][5]);
        }

    }
    fclose(stdin);
    return 0;
}