D. Olya and Energy Drinks【BFS变形】

D. Olya and Energy Drinks

题意：每次可以直着走1~k步，问从(x1,y1)出发到(x2,y2)至少需要多少步

思路：直接BFS，但有一点要注意的是，判断一点(nx,ny)是不是可以转移是判dis[nx][ny]>dis[x][y]。前一个写法是用vis标记，然而发现WA49.看了数据才知道，可能存在某种情况，使得真正的转移达不到。BFS貌似这样写，也比较好，用dis判转移

#include<bits/stdc++.h>
#define PI acos(-1.0)
#define pb push_back
#define F first
#define S second
#define debug puts
#define setp cout << fixed << setprecision(3)
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
typedef long long ll;
const int N=1e3+3;
const int MOD=1e9+7;
const ll INF=1e18+8;
int  n,m,k,sx,sy,ex,ey;
char mp[N][N];
int  dis[N][N];
int dir[4][2]={1,0,-1,0,0,-1,0,1};

bool bfs(){
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)   dis[i][j]=1e9+9;
    queue<int> q;
    q.push(sx);
    q.push(sy);
    dis[sx][sy]=0;
    while(!q.empty()){
        int x=q.front();q.pop();
        int y=q.front();q.pop();
//        cout <<"~"<<x<<" "<<y<<endl;
        if(x==ex&&y==ey){
            cout << dis[x][y]<<endl;
            return true;
        }
        for(int i=0;i<4;i++){
            for(int t=1;t<=k;t++){
                int nx=x+dir[i][0]*t;
                int ny=y+dir[i][1]*t;

                if(nx<1||nx>n||ny<1||ny>m||mp[nx][ny]!='.'||dis[nx][ny]<=dis[x][y])  break;
                if(dis[nx][ny]==dis[x][y]+1)    continue;///不这样写会MLE，毕竟压入的点太多了。这样保证最多n*m个点
                dis[nx][ny]=dis[x][y]+1;
//                cout <<"`"<<nx<<" "<<ny<<endl;
                q.push(nx);
                q.push(ny);
            }
        }
//        for(int y=b+1;y<=m&&y<=b+k&&mp[a][y]=='.'&&dis[a][y]>dis[a][b];y++)
//            q.push(a),q.push(y),/*cout<<a<<" "<<y<<endl,*/dis[a][y]=dis[a][b]+1;
//        for(int y=b-1;y>=1&&y>=b-k&&mp[a][y]=='.'&&dis[a][y]>dis[a][b];y--)
//            q.push(a),q.push(y),/*cout<<a<<" "<<y<<endl,*/dis[a][y]=dis[a][b]+1;
//        for(int x=a+1;x<=n&&x<=a+k&&mp[x][b]=='.'&&dis[x][b]>dis[a][b];x++)
//            q.push(x),q.push(b),/*cout<<x<<" "<<b<<endl,*/dis[x][b]=dis[a][b]+1;
//        for(int x=a-1;x>=1&&x>=a-k&&mp[x][b]=='.'&&dis[x][b]>dis[a][b];x--)
//            q.push(x),q.push(b),/*cout<<x<<" "<<b<<endl,*/dis[x][b]=dis[a][b]+1;
//        puts("***********");
    }
    return false;
}


int main(void){
    FAST_IO;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
        cin>>mp[i]+1;
    cin >> sx >> sy >>ex >> ey;
    if(sx==ex&&sy==ey){
        cout << 0 << endl;
        return 0;
    }
    bool flag=bfs();
    if(!flag)   cout <<-1<<"\n";
    return 0;
}