poj 1222 开关问题+二进制枚举

点击打开链接

//二进制枚举+开关问题,变为全0串
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int a[5][6];
int dx[]={0,-1,1,0,0};          //五方向转移
int dy[]={0,0,0,-1,1};
int state[5][6],done[5][6],cnt;
void filp(int x,int y)
{
    done[x][y]=1;
    for(int i=0;i<5;i++)
    {
        int nx=x+dx[i],ny=y+dy[i];
        if(nx>=0&&nx<5&&ny>=0&&ny<6) state[nx][ny]=!state[nx][ny];
    }
}
bool is_zero()
{
    for(int i=0;i<6;i++)
        if(state[4][i]) return false;
    return true;
}
int main()
{
    int T,kase=0;
    scanf("%d",&T);
    while(T--)
    {
        for(int i=0;i<5;i++)
            for(int j=0;j<6;j++)
                 scanf("%d",&a[i][j]);
        printf("PUZZLE #%d\n",++kase);
        int ans[5][6],pos=50;
        for(int i=0;i<(1<<6);i++)
        {
            memcpy(state,a,sizeof(a));
            memset(done,0,sizeof(done));
            cnt=0;
            int t=5;
            for(int j=0;j<6;j++)
            {
                if((i>>j)&1) {
                    cnt++;
                    filp(0,t);
                }
                t--;
            }
            for(int i=1;i<5;i++)
            {
                for(int j=0;j<6;j++)
                     if(state[i-1][j]) {
                       cnt++;
                       filp(i,j);
                  }
            }
            if(is_zero()&&cnt<pos) {
                pos=cnt;
                memcpy(ans,done,sizeof(done));
            }
        }
        for(int i=0;i<5;i++)
        {
            for(int j=0;j<5;j++)
                printf("%d ",ans[i][j]);
            printf("%d\n",ans[i][5]);
        }
    }
    return 0;
}