POJ - 3279 Fliptile

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <queue>
using namespace std;
const int maxn=20;
const int INF=0x3f3f3f3f;
int m,n;
int a[maxn][maxn];
int Min=INF;
int op[20][20];
int ans[maxn][maxn];
int loc[5][2]={{0,0},{1,0},{-1,0},{0,1},{0,-1}};
int judge (int posx,int posy)
{
    int t=a[posx][posy];
    for (int i=0;i<5;i++)
    {
        int x=posx+loc[i][0];
        int y=posy+loc[i][1];
        if(x>=0&&x<m&&y>=0&&y<n) t+=op[x][y];
    }
    return t&1;
}
int enu ()
{
    int tmin=0;
    for (int i=1;i<m;i++)
    {
        for (int j=0;j<n;j++)
        {
            if(judge(i-1,j)) op[i][j]=1;
        }
    }
    for (int i=0;i<n;i++)
    {
        if(judge(m-1,i)) return INF;
    }
    for (int i=0;i<m;i++)
    {
        for (int j=0;j<n;j++)
            tmin+=op[i][j];
    }
    return tmin;
}
int main()
{
    memset (ans,0,sizeof(ans));
    scanf("%d%d",&m,&n);
    for (int i=0;i<m;i++)
    {
        for (int j=0;j<n;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    for (int i=0;i<1<<n;i++)
    {
        memset (op,0,sizeof(op));
        for (int j=0;j<n;j++)
            op[0][j]=i>>j&1;
        int tmin=enu();
        if(tmin<Min)
        {
            Min=tmin;
            memcpy(ans,op,sizeof(op));
        }
    }
    if(Min==INF)
    {
        printf("IMPOSSIBLE\n");
        return 0;
    }
    for (int i=0;i<m;i++)
    {
        for (int j=0;j<n;j++)
        {
            printf("%d",ans[i][j]);
            if(j==n-1) printf("\n");
            else printf(" ");
        }
    }
    return 0;
}