【BFS+剪枝优化】Codeforces Round #442 (Div. 2) D. Olya and Energy Drinks

Codeforces Round #442 (Div. 2) D. Olya and Energy Drinks
题意：地图上只有可走格和不可走格，我们从(sx,sy)走到(ex,ey)，每秒最多走k步，问最少需要多少时间。

思路：

1. 就是bfs跑，但是要注意进了队的点是否走过一个格，是一定不能影响别的点是不是选择走这一格的，所以不能直接入队就vis[][]=1标记之后别的点就不能再跑这一点了，这样可能损失最优解。
2. 所以我们现在用vis[][]来表示起点走到这一点的最短时间，兼顾了标记该点走过的作用。
3. 这里可以做一个剪枝(不做的话基本就是950ms+卡过)：我们想，同一个方向上的点（先暂时不考虑障碍，假设都可以走），那么我们从s点最多从1走到k，这些点我们一定是push进队了的，所以拿出来更新其他点的时候，我们就可以考虑了：1来更新的话，是更新不了2到k的，因已经被s更新过了。有人可能会问，那如果这里直接break的话，点k+1怎么办？（我也纠结了很久）其实k+1可以由k来更新。到k更新的时候，会更新k+1到2k的点，所以不会有遗漏的点。 如果有障碍或者到了边界呢，这样的话就在不到k的时候就直接break了。【当然了这只能做单条边方向上的优化，不是对一个点做完了所有四个方向上的优化】
4. 剩下的话就是常规的bfs了。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <list>
#include <map>
#define P(x) x>0?x:0
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;
typedef vector<int>:: iterator VITer;
const int maxn=1e3+5;
const int dir[4][2]={
        1,0,
        0,1,
        -1,0,
        0,-1
};
char mp[maxn][maxn];
int vis[maxn][maxn];
int sx,sy,ex,ey;
int n,m,k;
struct node
{
    int x, y, s;
    node(int a=0,int b=0,int c=0):x(a),y(b),s(c){}
};
int bfs()
{
    memset(vis,-1, sizeof(vis));
    queue<node >q;
    q.push(node(sx,sy,0));
    vis[sx][sy]=0;
    while(!q.empty())
    {
        node tmp=q.front();
        for(int i=0;i<4;i++)
        {
            for(int j=1;j<=k;j++)
            {
                int xx=tmp.x+j*dir[i][0];
                int yy=tmp.y+j*dir[i][1];
                if(xx==ex && yy==ey)
                    return tmp.s+1;
                if(xx<=0||yy<=0||xx>n||yy>m||mp[xx][yy]=='#')
                    break;
                if(vis[xx][yy]!=-1 && vis[xx][yy]<tmp.s+1)
                    break;
                if(vis[xx][yy]==-1)//vis[xx][yy]!=-1的话，说明已经push进队过了
                    q.push(node(xx,yy,tmp.s+1));
                vis[xx][yy]=tmp.s+1;
            }
        }
        q.pop();
    }
    return -1;
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        getchar();
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                scanf("%c",&mp[i][j]);
            }
            getchar();
        }
        scanf("%d%d%d%d",&sx,&sy,&ex,&ey);
        if(sx==ex && sy==ey)
        {
            printf("0\n");
            continue;
        }
        if(k>max(n,m))
            k=max(n,m);
        printf("%d\n",bfs());
    }
    return 0;
}