CodeForces - 793B（bfs）

题目

用四维来记录状态。
后面两个分别记录当前的转弯次数和从哪个方向来的。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=1e3+10;
char mp[maxn][maxn];
int n,m;
int starx,stary,endx,endy;
int vis[maxn][maxn][4][6];
struct node
{
	int x,y,step,id;
}s,t;
int nxt[5][3]={{0,1},{0,-1},{-1,0},{1,0}};
bool check(int x,int y,int z,int w)
{
	if(x>=0&&x<n&&y>=0&&y<m&&mp[x][y]!='*'&&!vis[x][y][z][w]&&z<=2)return true;
	else  return false;
}
bool bfs()
{
	queue<node>q;
	s.x=starx,s.y=stary,s.step=0,s.id=-1;
	vis[s.x][s.y][0][4]=1;
	q.push(s);
	while(!q.empty())
	{
		s=q.front(),q.pop();
		for(int i=0;i<4;i++)
		{
			t.x=s.x+nxt[i][0],t.y=s.y+nxt[i][1];
			if(s.id!=-1&&s.id!=i)t.step=s.step+1;
			else t.step=s.step;
			if(check(t.x, t.y, t.step, i))
			{
				//cout<<t.x<<" "<<t.y<<endl;
				vis[t.x][t.y][t.step][i]=1;
				if(t.x==endx && t.y==endy)return true;
				t.id=i;
				q.push(t);
			}
		}
	}
	return false;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<n;i++)
	{
		scanf("%s",mp[i]);
	}
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<m;j++)
		{
			if(mp[i][j]=='S')starx=i,stary=j;
			if(mp[i][j]=='T')endx=i,endy=j;
		}
	}
	if(bfs())printf("YES\n");
	else printf("NO\n");
	return 0;
}