Uva1600 - Patrol Robot（bfs解法）

题目大意：一个机器人要从左上角(1,1) 走到右下角(m, n) （当然读入m，n的时候因为习惯我反过来了），然后路上有障碍物， 但是机器人每次可以连续穿过最多k个障碍物——是每次，不是累计的，刷这题的时候好几个人同学理解错题意挂在这了。然后问所需要的最短时间。

看完这题第一反应就是bfs。。然后再考虑穿过障碍物的情况，由于地图较小（1 <= m, n <= 20）， 所以时间上不要虚，于是障碍物中我用了类似spfa的技巧， 也就是松弛操作，到达当前障碍物的步数（从起点开始记）如果小于之前记录的值就更新并且再次加入队列。现在隔了这么久发现其实用不上松弛。。毕竟按照bfs的尿性第一次访问到这个点的时候肯定是最短的。。管他呢反正A掉了就酱紫了吧Orz。

话不多说见代码：

#include <set>
#include <map>
#include <queue>
#include <stack>
#include <cmath>
#include <string>
#include <cctype>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <sstream>
#include <iostream>
#include <algorithm>
using namespace std;
#define ls id<<1,l,mid
#define rs id<<1|1,mid+1,r
#define OFF(x) memset(x,-1,sizeof x)
#define CLR(x) memset(x,0,sizeof x)
#define MEM(x) memset(x,0x3f,sizeof x)
#define rep(i,a,b) for (int i=a;i<b;i++)
#define Rep(i,a,b) for (int i=a;i>=b;i--)
typedef long long ll ;
typedef pair<int,int> pii ;
const int maxn = 1e5+50 ;
const int inf = 0x3f3f3f3f ;
const int MOD = 1e9+7 ;
int T,n,m,k;
int dir[4][2]={1,0, -1,0, 0,1, 0,-1};
int mp[50][50],vis[30][30],dis[30][30];
struct state{
    int x,y;
    int Step;
};
bool check(state a) {return a.x&&a.x<=n&&a.y&&a.y<=m ;}

int bfs(){
    state st;
    st.x=1,st.y=1,st.Step=0;
    queue<state>q;
    q.push(st);
    vis[1][1]=1;
    while (q.size()){
        state now=q.front(),nxt;q.pop();
        int x=now.x,y=now.y,step=now.Step;
        if (mp[x][y]) vis[x][y]=0;
        if (x==n&&y==m) return step;
        for (int i=0;i<4;i++) {
            nxt.x=x+dir[i][0];
            nxt.y=y+dir[i][1];
            nxt.Step=step+1;
            if (check(nxt)) {
                if (mp[nxt.x][nxt.y]) {
                    if (dis[nxt.x][nxt.y]>dis[x][y]+1&&dis[x][y]+1<=k) {
                        dis[nxt.x][nxt.y]=dis[x][y]+1;
                        if (!vis[nxt.x][nxt.y]) q.push(nxt);
                        vis[nxt.x][nxt.y]=1;
                    }
                } else {
                    if (!vis[nxt.x][nxt.y]) q.push(nxt);
                    vis[nxt.x][nxt.y]=1;
                }
            }
        }
    }
    return -1;
}


int main () {
#ifdef LOCAL
	freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif
    scanf("%d",&T);
    while (T--) {
        MEM(dis);
        CLR(vis);
        scanf("%d%d%d",&n,&m,&k);
        for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) {
            scanf("%d",&mp[i][j]);
            if (!mp[i][j]) dis[i][j]=0;
        }
        int ans=bfs();
        cout << ans << endl ;
    }
	return 0;
}