习题6-5 UVA1600 巡逻机器人 Patrol Robot

难度：4

稍微难一点的bfs，以前数组的bfs都是用二维，来记录这个点入过队没有，但是题目加了一个条件，那就是最多连续穿墙不超过一个数，那么这就又是一维状态，加一维来表示到这个点连续穿墙的次数，也就是题目貌似加了障碍，但这个障碍是一会儿可走一会儿不可走的，不像别的题，要么可走要么不可走，这个一开始你穿过的障碍物少的时候，是可以走的，等穿的墙多了，就变得不可走了，所以就是可能几次走到了同一点，但是穿墙次数不一样，这时属于不同的情况，所以开一维数组来表示这不同的情况。

然后需要想明白的很重要的一个点是，当前点的k，也就是连续穿墙次数，如果当前点不是障碍物，那么归零，如果是障碍物，那么由前一个点的k加一，以前是判断两个值，也就是会不会出界，现在加上一个要判断的值，

#include <bits/stdc++.h>

#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(), (x).end()

using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pa;

struct node {
    int x, y, k;
    node() {}
    node(int a, int b, int c): x(a), y(b), k(c) {}
};

int n, m, k;
int graph[30][30];
int vis[30][30][30];
int ans[30][30][30];
int dx[4] = {0, 1, 0, -1};
int dy[4] = {-1, 0, 1, 0};

bool judge(int x, int y, int kk) {
    if (x < 1 || x > n || y < 1 || y > m || kk > k) return false;
    return true;
}

int bfs() {
    memset(vis, 0, sizeof(vis));
    queue<node> q;
    q.push(node(1, 1, 0));
    vis[1][1][0] = 1;
    ans[1][1][0] = 0;
    while(!q.empty()) {
        node now = q.front();
        q.pop();
        if (now.x == n && now.y == m) return ans[now.x][now.y][now.k];
        for (int i = 0; i < 4; i++) {
            int xx = now.x + dx[i];
            int yy = now.y + dy[i];
            int kk;
            if (graph[xx][yy]) kk = now.k + 1;
            else kk = 0;
            if (judge(xx, yy, kk) && !vis[xx][yy][kk]) {
                q.push(node(xx, yy, kk));
                vis[xx][yy][kk] = 1;
                ans[xx][yy][kk] = ans[now.x][now.y][now.k] + 1;
            }
        }
    }
    return -1;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        cin >> n >> m >> k;
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                cin >> graph[i][j];
            }
        }
        cout << bfs() << endl;
    }
    return 0;
}