codeforces 877 problem D Olya and Energy Drinks bfs剪枝

傳送門
//題意: 就是給定一個起點和一個終點, 每次可以往四個方向最多走k步, 每次走花費一秒, 問走到終點最少需要多少秒.
//思路: 直接bfs肯定會T, 我們需要加上幾個剪枝, 一個是當我們走到某點時, 該點曾經被到達過，並且此時的時間比從當前點走到這個點的時間還要短, 那麼就不要走了, 因為後續的點由這個點到達更優一點. 具體細節請看代碼.

AC Code

const int maxn = 1e3+5;int cas=1;char mapp[maxn][maxn];int vis[maxn][maxn],dis[maxn][maxn];int dx[] = {1,-1,0,0};int dy[] = {0,0,1,-1};void solve(){    int n,m,k;    int sx,sy,ex,ey;    while(~scanf("%d%d%d",&n,&m,&k)){        for(int i=1;i<=n;i++){            scanf("%s",mapp[i]+1);        }        scanf("%d%d%d%d",&sx,&sy,&ex,&ey);        Fill(vis,0); Fill(dis,inf);        queue<pair<int,int>>q;        q.push(make_pair(sx,sy));        vis[sx][sy] = 1;        dis[sx][sy] = 0;        while(!q.empty()){            pair<int,int>tmp = q.front();            q.pop();            for(int i=0;i<4;i++){                for(int j=1;j<=k;j++){                    int xx = tmp.fi+dx[i]*j;                    int yy = tmp.se+dy[i]*j;                    if(xx<1 || xx>n || yy<1 || yy>m) break;                    if(mapp[xx][yy] == '#') break;                    if(dis[xx][yy]!=inf && dis[xx][yy] < dis[tmp.fi][tmp.se] + 1) break;                    if(vis[xx][yy]) continue;                    vis[xx][yy] = 1;                    dis[xx][yy] = dis[tmp.fi][tmp.se] + 1;                    q.push(make_pair(xx,yy));                }            }        }        if(dis[ex][ey]!=inf) cout << dis[ex][ey] << endl;        else cout << "-1" << endl;    }}