【BZOJ 1054】 [HAOI2008]移动玩具 bfs

水一下

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
int vis[1000020],t,a[5][5];
queue<pair<int,int> >q;
const int dx[]={0,0,1,-1};
const int dy[]={1,-1,0,0};
const int e[5][5]={{0,0,0,0,0},{0,1,2,4,8},{0,16,32,64,128},{0,256,512,1024,2048},{0,4096,8192,16384,32768}};

void bfs(){
	while(!q.empty()){
		int u=q.front().first,ste=q.front().second;q.pop();
		
		for(int i=1;i<=4;i++){
			for(int j=0;j<4;j++){
				a[i][j+1]= (u&(1<<((i-1)*4+j)) )? 1 : 0;
			}
		}
		
		int nx,ny,x;
		for(int i=1;i<=4;i++){
			for(int j=1;j<=4;j++){
				if(a[i][j]){
					for(int k=0;k<4;k++){
						nx=i+dx[k],ny=j+dy[k];
						if(nx>4||nx<1||ny>4||ny<1||a[nx][ny])continue;
						x=u+e[nx][ny]-e[i][j];
						if(vis[x])continue;
						if(x==t){printf("%d",ste+1);return;}
						vis[x]=1;q.push(make_pair(x,ste+1)); 
					}
				}
			}
		}
		
	}
}

int main(){
	int st=0;char s[15];
	for(int i=1;i<=4;i++){
		scanf("%s",s);
		for(int j=0;j<4;j++){
			st|=(s[j]-'0')<<((i-1)*4+j);
		}
	}q.push(make_pair(st,0));
	for(int i=1;i<=4;i++){
		scanf("%s",s);
		for(int j=0;j<4;j++){
			t|=(s[j]-'0')<<((i-1)*4+j);
		}
	}
	if(st==t){puts("0");return 0;}
	bfs();
	return 0;
}