[各种面试题] 黑白迷阵

题目描述：

黑白迷阵是一个GrassLand编写的手机游戏，它的规则非常简单，有如下4*5的棋盘，其中一些是格子是黑色，一些格子是白色的，每当点击其中某一个格子，它以及它上下左右五个格子的颜色会发出反转，如下图

                     

           

游戏胜利的条件很简单，把所有的格子变为黑色即可。
GrassLand想知道，给定一个游戏格局，至少需要几次点击，就可以获得游戏的胜利。

输入：

输入包含多组测试用例。输入的第一行为一个整数T，代表共有的测试用例数。紧接着为T组测试用例。
每组测试用例，为由四行五列01阵列表示的游戏格局，其中0代表该格子的颜色为白色，1代表该格子的颜色为黑色。

输出：

对于每组测试用例，输出为一个整数，为至少需要的点击次数。数据保证所给格局可以在有限步内取得胜利。

样例输入：

20011101111111111111100111011111111011100

样例输出：

12

一个比较直观的BFS，重点是要用一个int来表示一个棋盘状态。然后用BFS求解。

我这个直接TLE了，懒得改了，思路反正是对的。

#include<iostream>#include<cstdio>#include<vector>#include<string>#include<cstring>#include<climits>#include<queue>#include<map>#include<algorithm>using namespace std;const int HEIGHT=4;const int WIDTH=5;const int END=(1<<20)-1;int used[END+10];int step[END+10];char s[10];void roll(int x,int y,int& grid){if(x<0||x>=HEIGHT||y<0||y>=WIDTH)return;int k=(x*WIDTH)+y;grid=grid^(1<<k);}int click(int x,int y,int grid){int a[]={0,-1,+1,0,0};int b[]={0,0,0,-1,+1};for(int i=0;i<5;i++)roll(x+a[i],y+b[i],grid);return grid;}int init(){int k=0;int grid=0;for(int i=0;i<4;i++){scanf("%s",s);for(int j=0;j<5;j++){grid|=(s[j]-'0')<<k;k++;}}return grid;}int solve(){int grid=init();queue<int> Q;Q.push(grid);memset(used,0,sizeof(used));used[grid]=1;memset(step,0,sizeof(step));while(!Q.empty()){int now=Q.front();Q.pop();if(now==END){return step[now];}for(int x=0;x<4;x++){for(int y=0;y<5;y++){int state=click(x,y,now);if(!used[state]){used[state]=1;step[state]=step[now]+1;Q.push(state);}}}}return 0;}int main(){int t;scanf("%d",&t);while(t--){int ret=solve();printf("%d\n",ret);}}