AcWing 172. 立体推箱子 BFS+状态表示

Posted 2021-11-11 karshey

题
代码参考了书上的。判断是否合法的函数写的好精简！

状态表示的理解：

//lie==0 立着 lie==1 横着躺着  lie==2 竖着躺着
//j:0123分别表示左右上下
//nextx[i][j]代表lie==i时x往j方向走的变化
const int nextx[3][4]={{0,0,-2,1},{0,0,-1,1},{0,0,-1,2}};
const int nexty[3][4]={{-2,1,0,0},{-1,2,0,0},{-1,1,0,0}};
const int nextlie[3][4]={{1,1,2,2},{0,0,1,1},{2,2,0,0}};

比如：
立着时想往左走，即[0][0]，走法就是变成横躺，左上角的坐标：行不变，列-2。
因此：
nextx[0][2]=0;//行不变即x不变
nexty[0][2]=-2;//列-2即对应的值为-2
nextlie[0][2]=1;//1表示横躺
正好对应。

总代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define debug(x) cout<<#x<<" "<<x<<endl;
const int inf=0x3f3f3f3f;
const int MOD=1e9+7;
//书上代码写的太好了 所以对着打一打======================
const int N=550;
char g[N][N];
int d[N][N][3];
struct node
{
	int x,y,lie;
};
node s,e;
queue<node>q;
int n,m;
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
void startandend()
{
	for(int i=1;i<=n;i++) scanf("%s",g[i]+1);
	
	int flag=0;
	fir(i,1,n)
		fir(j,1,m)
		{
			if(g[i][j]=='O') 
			{
				e.x=i,e.y=j,e.lie=0;
				g[i][j]='.';
				flag++;
			}
			else if(g[i][j]=='X')
			{
				int x1=i,y1=j;
				int x2=i,y2=j;
				g[i][j]='.';
				flag++;
				s={x1,y1,0};
				for(int k=0;k<4;k++)
				{
					if(g[i+dx[k]][j+dy[k]]=='X')
					{
						x2=i+dx[k];
						y2=j+dy[k];
						g[x2][y2]='.';						
						if(x1==x2)//heng
						{
							y1=min(y1,y2);
							s={x1,y1,1};
						}
						else
						{
							x1=min(x1,x2);
							s={x1,y1,2};
						}						
						break;
					}
				}							
			}
			if(flag==2) break;	
		}				
}

int judge_fw(int x,int y)
{
	return x>=1&&x<=n&&y>=1&&y<=m;
}
int judge(node t)
{
	if(!judge_fw(t.x,t.y)) return 0;
	if(g[t.x][t.y]=='#') return 0;
	
	if(t.lie==0&&g[t.x][t.y]!='.') return 0;
	if(t.lie==1&&g[t.x][t.y+1]=='#') return 0;
	if(t.lie==2&&g[t.x+1][t.y]=='#') return 0;
	
	return 1;
}
const int nextx[3][4]={{0,0,-2,1},{0,0,-1,1},{0,0,-1,2}};
const int nexty[3][4]={{-2,1,0,0},{-1,2,0,0},{-1,1,0,0}};
const int nextlie[3][4]={{1,1,2,2},{0,0,1,1},{2,2,0,0}};
void bfs()
{
	while(q.size()) q.pop();//clear
	
	fir(i,1,n)
		fir(j,1,m)
			for(int k=0;k<3;k++)
			{
				d[i][j][k]=-1;
			}
	
	d[s.x][s.y][s.lie]=0;
	q.push(s);
	while(q.size())
	{
		node now=q.front();
		q.pop();
		
		for(int i=0;i<4;i++)
		{
			node next={now.x+nextx[now.lie][i],now.y+nexty[now.lie][i],nextlie[now.lie][i]};
			if(!judge(next)) continue;
			
			if(d[next.x][next.y][next.lie]==-1)
			{
				d[next.x][next.y][next.lie]=d[now.x][now.y][now.lie]+1;
				q.push(next);
			}
		}
	}
}
int main()
{
	while(cin>>n>>m&&n&&m)
	{
		startandend();
		bfs();
		if(d[e.x][e.y][e.lie]==-1) cout<<"Impossible";
		else cout<<d[e.x][e.y][e.lie];
		cout<<endl;
	}		
	return 0; 
}