bzoj 1644: [Usaco2007 Oct]Obstacle Course 障碍训练课【spfa】

洛谷的数据毒啊

把(i,j,k)作为一个点spfa，表示点(i,j)朝向k方向，然后向四个方向转移即可

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int N=105,inf=1e9,dx[]={-1,1,0,0},dy[]={0,0,-1,1};
int n,sx,sy,tx,ty,dis[N][N][5];
char a[N][N];
bool v[N][N][5];
struct qwe
{
	int x,y,k;
	qwe(int X=0,int Y=0,int K=0)
	{
		x=X,y=Y,k=K;
	}
};
inline bool ok(int x,int y)
{
	return x>=1&&x<=n&&y>=1&&y<=n&&a[x][y]!='x';
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%s",a[i]+1);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			if(a[i][j]=='A')
				sx=i,sy=j;
			if(a[i][j]=='B')
				tx=i,ty=j;
			for(int k=0;k<4;k++)
				dis[i][j][k]=inf;
		}//cerr<<sx<<" "<<sy<<"   "<<tx<<" "<<ty<<endl;
	queue<qwe>q;
	v[sx][sy][0]=1,v[sx][sy][1]=1,v[sx][sy][2]=1,v[sx][sy][3]=1;
	dis[sx][sy][0]=0,dis[sx][sy][1]=0,dis[sx][sy][2]=0,dis[sx][sy][3]=0;
	q.push(qwe(sx,sy,0)),q.push(qwe(sx,sy,1)),q.push(qwe(sx,sy,2)),q.push(qwe(sx,sy,3));
	while(!q.empty())
	{
		int x=q.front().x,y=q.front().y,k=q.front().k;
		q.pop();
		v[x][y][k]=0;
		for(int i=0;i<4;i++)
			if(ok(x+dx[i],y+dy[i])&&dis[x+dx[i]][y+dy[i]][i]>dis[x][y][k]+(k!=i))
			{
				dis[x+dx[i]][y+dy[i]][i]=dis[x][y][k]+(k!=i);
				if(!v[x+dx[i]][y+dy[i]][i])
				{
					v[x+dx[i]][y+dy[i]][i]=1;
					q.push(qwe(x+dx[i],y+dy[i],i));
				}
			}
	}
	// for(int i=1;i<=n;i++)
		// for(int j=1;j<=n;j++)
			// for(int k=0;k<4;k++)
				// cerr<<i<<" "<<j<<" "<<k<<" "<<dis[i][j][k]<<endl;
	printf("%d\n",min(min(dis[tx][ty][0],dis[tx][ty][1]),min(dis[tx][ty][2],dis[tx][ty][3])));
	return 0;
}
/*
3
.xA
...
Bx.
*/