POJ3322 Bloxorz I

题目链接

这道题做了一天...纪念一下。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
struct node 
{
	int x,y,sta;
	node(int a,int b,int c): x(a),y(b),sta(c) {}
};
int n,m,endx,endy;
char a[510][510];
int book[510][510][3];
const int nxtx[3][4]={{-2,0,1,0},{-1,0,1,0},{-1,0,2,0}};
const int nxty[3][4]={{0,-2,0,1},{0,-1,0,2},{0,-1,0,1}};
const int nxtz[3][4]={{2,1,2,1},{1,0,1,0},{0,2,0,2}};
bool in(int x,int y) {return x>=1 && x<=n && y>=1 && y<=m;} 
bool check(node s)
{
	int x=s.x,y=s.y,sta=s.sta;
	if(a[x][y]=='#') return false;
	if(!in(x,y)) return false;
	if(~book[x][y][sta]) return false;
	if(sta==0)
	{
		if(a[x][y]=='E') return false;
		return true;
	}
	if(sta==1)
	{
		if(a[x][y+1]=='#') return false;
		if(!in(x,y+1)) return false;
		return true;
	}
	if(sta==2)
	{
		if(a[x+1][y]=='#') return false;
		if(!in(x+1,y)) return false;
		return true;
	}
	return true;
}
int bfs(int stx,int sty,int ststa)
{
	queue<node> que;
	que.push(node(stx,sty,ststa));
	book[stx][sty][ststa]=0;
	while(!que.empty())
	{
		node head=que.front(); que.pop();
		int x=head.x,y=head.y,sta=head.sta;
		int stp=book[x][y][sta]+1;
		for(int i=0;i<4;i++)
		{
			int tx=x+nxtx[sta][i];
			int ty=y+nxty[sta][i];
			int tsta=nxtz[sta][i];
			if(!check(node(tx,ty,tsta))) continue;
			book[tx][ty][tsta]=stp;
			que.push(node(tx,ty,tsta));
			if(tx==endx && ty==endy && tsta==0) return book[tx][ty][tsta];
		}
	}
	return -1;
}
// 0/1/2 竖/橫躺/竖躺 
int main()
{
	while(~scanf("%d %d",&n,&m) && n)
	{
		endx=endy=0;
		memset(a,0,sizeof(a));
		memset(book,-1,sizeof(book));
		for(int i=1;i<=n;i++) scanf("%s",a[i]+1);
		int ststa=0,stx=0,sty=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				if(a[i][j]=='X') 
				{
					a[i][j]='.';
					stx=i;
					sty=j;
					if(a[i+1][j]=='X') 
					{
						ststa=2;
						a[i+1][j]='.';
					} 
					else if(a[i][j+1]=='X') 
					{
						ststa=1;
						a[i][j+1]='.'; 
					}
					else ststa=0;
//					break;
				} 
				if(a[i][j]=='O')
				{
					endx=i;
					endy=j;
					a[i][j]='.';
				}
			}
		}
		int kkksc03=bfs(stx,sty,ststa);
		if(kkksc03!=-1) printf("%d\n",kkksc03);
		else puts("Impossible");
	}
	return 0;
}