F: Pond Skater(BFS)

F: Pond Skater(BFS)

传送门
思路:四个方向 B F S BFS ,开一个二维数组保存到达该位置的最小步数。

遇到不是最优的情况就进行剪枝,不断更新最值。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5,M=1e6+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a) memset(a,0,sizeof a)
#define lx x<<1
#define rx x<<1|1
#define reg register
#define PII pair<int,int>
#define fi first 
#define se second
int H,W,K;
int d[4][2]={0,1,0,-1,1,0,-1,0};
int main(){
	scanf("%d%d%d",&H,&W,&K);
	vector<string>S(H);
	int sx,sy,ex,ey;
	scanf("%d%d%d%d",&sx,&sy,&ex,&ey);
	sx--,sy--,ex--,ey--;
	for(int i=0;i<H;i++) cin>>S[i];
	vector<vector<int> >a(H,vector<int>(W,1e9));
	queue<PII>q;
	a[sx][sy]=0;
	q.push({sx,sy});
	while(!q.empty()){
		 int x=q.front().fi,y=q.front().se;
		 q.pop();
		 for(int i=0;i<4;i++){
		 	 int nx=x,ny=y;
		 	 int tmp=a[x][y]+1;
		 	 for(int k=1;k<=K;k++){
		 	 	  nx+=d[i][0],ny+=d[i][1];
		 	 	  if(nx<0||nx>=H||ny<0||ny>=W||S[nx][ny]=='@'||a[nx][ny]<tmp) break;//剪枝. 
		 	 	  if(a[nx][ny]>tmp){	//更新状态. 
		 	 	  		a[nx][ny]=tmp;
		 	 	  		q.push({nx,ny}); 
					}
			  }
		 }
	}
	printf(a[ex][ey]==1e9?"-1":"%d\n",a[ex][ey]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_45750972/article/details/106764836
今日推荐