POJ 1222 EXTENDED LIGHTS OUT 开关问题

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

Problem: 1222       User: ChenyangDuMemory: 156K        Time: 0MSLanguage: C++       Result: AcceptedSource 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;}