解题思路:如果枚举每一个格子的开关,复杂度为O(2^MN),这会超时。但观察规律可知,只要第一行的开关状态确定,后面的状态也都确定了。由此枚举第一行的开关状态即可。这样复杂度变为O(MN2^N)。
#include<cstdio>
#include<cstring>
#include<iostream>
#define inf 0x3f3f3f3f
using namespace std;
int s[20][20];
int dy[5]={-1,0,0,1};
int dx[5]={0,-1,0,0};
int flip[20][20],fb[20][20];
int m,n;
bool judge(int x,int y)
{
int c=s[x][y];
for(int i=0;i<4;i++)
{
if(x+dx[i]>-1&&x+dx[i]<m&&y+dy[i]>-1&&y+dy[i]<n)
{
c+=flip[x+dx[i]][y+dy[i]];
}
}
if(c&1) return true;
return false;
}
int main()
{
//freopen("t.txt","r",stdin);
while(scanf("%d%d",&m,&n)!=EOF)
{
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
scanf("%d",&s[i][j]);
}
}
int res=inf;
for(int i=0;i<1<<n;i++)
{
memset(flip,0,sizeof(flip));
for(int j=0;j<n;j++)
{
flip[0][j]=(i>>j)&1;
}
for(int j=1;j<m;j++)
{
for(int k=0;k<n;k++)
{
if(judge(j-1,k))
flip[j][k]=1;
}
}
bool fg=false;
for(int j=0;j<n;j++)
{
if(judge(m-1,j))
{
fg=true;
break;
}
}
if(fg) continue;
int tp=0;
for(int j=0;j<m;j++)
{
for(int k=0;k<n;k++)
{
if(flip[j][k])
tp++;
}
}
if(res>tp)
{
memcpy(fb,flip,sizeof(flip));
res=tp;
}
}
if(res==inf)
printf("IMPOSSIBLE\n");
else
{
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
printf("%d%c",fb[i][j],j==n-1? '\n':' ');
}
}
}
}
}