https://vjudge.net/problem/POJ-3279#author=0
先说一下整体思路,先枚举第一行的2^(行数)种情况,然后往下深搜,实际上,第一行的状态确定了,下面行的状态也随之确定。最后判断最后一行是不是都是白色即0,否,则该情况不行。还有,发现个问题,系统的测试数据中可能没有针对字典序要求的用例。
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=17;
///描述某个点周围的四个点和它本身
int dir[5][2]= {{0,0},{0,1},{1,0},{0,-1},{-1,0}};
///Map存放棋盘,turn存放每个棋子(地板)翻转的状态(翻转几次)
///ans存放最后要输出的每个地板翻转的次数,cnt记录翻转地板的总次数
///res存放最小的翻转次数,M代表行数,N代表列数
int Map[maxn][maxn],turn[maxn][maxn],ans[maxn][maxn],cnt,res,M,N;
///获取当前坐标地板的颜色,0--白色,1--黑色
///这儿还有一种方法,不写getcolor,再定义一个cur[maxn][maxn]来保存当前的地板颜色
///getcolor实际上是通过turn来推断地板的颜色
int getColor(int x,int y)
{
int temp=Map[x][y];
for(int i=0; i<5; i++)
{
int xi=x+dir[i][0];
int yi=y+dir[i][1];
///因为(x,y)受到周围地板及自己翻转的影响
if(xi>=0 && xi<M && yi>=0 && yi<N) temp+=turn[xi][yi];
}
return temp%2;
}
///从第二层一直搜索到第M层
void DFS()
{
for(int i=1; i<M; i++)
{
for(int j=0; j<N; j++)
{
if(getColor(i-1,j)) turn[i][j]=1,cnt++; ///turn[i][j]=1,1表示要翻转,0不翻转
if(cnt>res) return; ///这里可以进行一点优化(剪枝)
}
}
///下面检测最后一层是否都是白色
for(int i=0; i<N; i++)
if(getColor(M-1,i)) return;
///更新ans
if(cnt<res) memcpy(ans,turn,sizeof(turn)),res=cnt;
}
int main()
{
while(cin>>M>>N) ///不写循环也可以,系统既可以单例也可以多例测试
{
res=INT_MAX;
for(int i=0; i<M; i++)
for(int j=0; j<N; j++)
cin>>Map[i][j];
///利用二进制位移来枚举第一行翻转的所有情况,
for(int i=0; i< 1<<N; i++) ///i<2^N
{
cnt=0;
memset(turn,0,sizeof(turn));
for(int j=0; j<N; j++)
{
turn[0][N-1-j]=i>>j&1; ///从后往前填,符合字典序从小到大
if(turn[0][N-1-j]) cnt++;
///但是从前往后填,即把N-1-j替换为j,也可以AC,故我怀疑测试数据中没有考虑字典序的用例
}
DFS();
}
if(res==INT_MAX) cout<<"IMPOSSIBLE"<<endl;
else
{
for(int i=0; i<M; i++)
{
for(int j=0; j<N; j++)
{
if(j>0) cout<<' ';
cout<<ans[i][j];
}
cout<<endl;
}
}
}
return 0;
}