Codeforces Round #442 (Div. 2) D. Olya and Energy Drinks（BFS+剪枝）好题

题目链接


题意：要你判断从起点到终点的最小步数，每次可以从上下左右四个方向沿着同一方向最多走k部。
思路：标准的bfs走迷宫问题，可以略有不同的是每次可以沿同一方向最多走k步，直接bfs的话会tle，考虑一下剪枝，那怎么剪呢？我们可以设一个d数组表示从起点到终点的最小步数，设当前点为（x，y）可以沿着右方向（x，y+1),(x,y+2).。。。(x,y+k)，假设我们求出d【x】【y】+1>d【x】【y+2】那么后面的（x，y+3），（x，y+4）。。。（x，y+k）其实都可以不用走了，都是多余的，因为存在着比（x，y）更优的方案。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e3+5;
const int dir[4][2]={{1,0},{-1,0},{0,1},{0,-1}};
const int inf=0x3f3f3f3f;
int x1,y1,x2,y2,n,m,k,vis[maxn][maxn],d[maxn][maxn];
char s[maxn][maxn];
bool check(int x,int y)
{
	return (x>=1&&x<=n&&y>=1&&y<=m);
}
void bfs()
{
	queue<pair<int,int>>q;
	q.push({x1,y1});
	vis[x1][y1]=1;
	d[x1][y1]=0;
	while(!q.empty())
	{
		pair <int,int>t=q.front();
		q.pop();
		int tx=t.first,ty=t.second;
		for(int i=0;i<4;++i)
		{
			int nx=tx,ny=ty;
			for(int j=1;j<=k;++j)
			{
				nx+=dir[i][0],ny+=dir[i][1];
				if(check(nx,ny)&&s[nx][ny]=='.'&&d[nx][ny]>=d[tx][ty]+1)
				{
					d[nx][ny]=d[tx][ty]+1;
					if(!vis[nx][ny]) q.push({nx,ny}),vis[nx][ny]=1;
				}
				else break;
			}
		}
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&k);
	memset(d,inf,sizeof(d));
	for(int i=1;i<=n;++i) scanf("%s",s[i]+1);
	scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
		bfs();
	printf("%d\n",d[x2][y2]==inf?-1:d[x2][y2]);
 }