题意:给你一个n*m的棋盘,有的棋子正面向上(0),有的棋子反面向上(1),问最少需要多少遍才能把所有的棋子变成正面向上?按字典序输出每个位置翻转的次数。
这题刚开始想的时候,是想着dfs的,毕竟出现在dfs的专题里面,原本的想法是记录下为1的位置,那么翻肯定就要在它们的附近翻,然后遍历所有的翻法,可是在写的过程发现,这种写法最后很难判断是否存在解,而且复杂度也很高。后来看了看大神的代码,恍然大悟。
思路:对于一个棋子,只有两种情况,正面向上和反面向上,并且操作两次的会回到初始状态,所以,对于每一个位置的棋子最多只需要操作一次。接着就是关键了:因为如果直接暴力枚举棋盘的所有情况需要2^(m*n),这很明显会TLE,于是我们只枚举第一行的所有情况。
具体操作:1、利用二进制子集枚举第一行n个位置的所有情况。对应位置如果是·1则反转对应位置。
2、对于2...m行,如果上行图对应的位置为1则对其下一行对应位置反转。(翻过的不翻,防止重复,所以能影响到的只有下一行)
3、最后判断一下最后一行是否能达到全面正面朝上,如果可以比较得出最短的步骤。
嗯...最后学习了memcpy来复制数组!
#include<cstdio>
#include<cstring>
#include<iostream>
#define inf 0x3f3f3f3f
using namespace std;
int m,n;
int maze[50][50],cntmaze[50][50],temp[50][50],cnt,mins,minmaze[50][50];
int dx[]={1,-1,0,0,0},dy[]={0,0,1,-1,0};
bool ok(){
for(int i=0;i<n;i++)
if(temp[m-1][i]==1)
return false;
return true;
}
void flip(int x,int y){
cntmaze[x][y]=1;
for(int i=0;i<5;i++){
int xx=x+dx[i];
int yy=y+dy[i];
if(xx>=0&&x<m&&yy>=0&&yy<n){
temp[xx][yy]=!temp[xx][yy];
}
}
}
void solve(int s){
cnt=0;
memcpy(temp,maze,sizeof(maze));
memset(cntmaze,0,sizeof(cntmaze));
for(int i=0;i<n;i++)
if((s>>i)&1){///找到要反转的位置
flip(0,i);
cnt++;
}
for(int i=1;i<m;i++)
for(int j=0;j<n;j++){
if(temp[i-1][j]==1){
flip(i,j);
cnt++;
}
}
if(ok()&&cnt<mins){
mins=cnt;
memcpy(minmaze,cntmaze,sizeof(cntmaze));
}
}
int main(){
while(cin>>m){
cin>>n;
memset(maze,0,sizeof(maze));///存图
mins=inf;
for(int i=0;i<m;i++)
for(int j=0;j<n;j++){
int t;
cin>>t;
if(t==1){
maze[i][j]=1;
}
}
for(int i=0;i<(1<<n);i++)///二进制枚举第一行所有可能排列情况
solve(i);
if(mins==inf)
cout<<"IMPOSSIBLE"<<endl;
else
for(int i=0;i<m;i++)
for(int j=0;j<n;j++)
printf("%d%c",minmaze[i][j],j==n-1?'\n':' ');
}
return 0;
}