在m*n的地图上,有障碍物(1)和空地(0),机器人要通过走上下左右四个方向从左上角(1,1)走到右下角(n,m),但不能连续超过k个障碍,要求最短步数。
走到同一点,有两种同一步数的走法,但是到达这一点所破除的障碍数量却不同,在这里我们需要选择障碍物更少的,因为很有可能在以后破除障碍能更快到达终点。
#include<iostream> #include<queue> #include<cstring> using namespace std; const int maxn = 20 + 2; struct Node { int x, y; Node(int x = 0, int y = 0) :x(x), y(y) {} bool operator == (const Node& rhs) const { return x == rhs.x && y == rhs.y; } }; int m, n; int d[maxn][maxn]; Node p[maxn][maxn]; int type[maxn][maxn]; int power[maxn][maxn]; int dr[] = { -1,0,1,0 }; int dc[] = { 0,1,0,-1 }; bool in(int x,int y) { return 0 <= x && x <= m - 1 && 0 <= y && y <= n - 1; } int BFS(int k) { queue<Node> Q; Node u(0,0); Q.push(u); memset(d, -1, sizeof(d)); memset(power, 0, sizeof(power)); d[0][0] = 0; while (!Q.empty()) { u = Q.front(); Q.pop(); if (u.x == m - 1 && u.y == n - 1) return d[m - 1][n - 1]; for (int i = 0; i < 4; i++) { Node v(u.x + dr[i], u.y + dc[i]); //即使是已经经过的点,若此次路线所经过的障碍累计总数更少,也可能再次经过,这与其他题不同 if (in(v.x, v.y) && power[u.x][u.y] + type[v.x][v.y] <= k && ( d[v.x][v.y] < 0 || ( d[v.x][v.y] == d[u.x][u.y] + 1 && power[u.x][u.y] + type[v.x][v.y] < power[v.x][v.y] ) ) ) { p[v.x][v.y] = u; d[v.x][v.y] = d[u.x][u.y] + 1; if (!type[v.x][v.y]) power[v.x][v.y] = 0; else power[v.x][v.y] = power[u.x][u.y] + type[v.x][v.y]; Q.push(v); } } } return -1; } int main() { int T; cin >> T; while (T--) { cin >> m >> n; int k; cin >> k; for (int i = 0; i < m; i++) for (int j = 0; j < n; j++) { cin >> type[i][j]; p[i][j] = Node(i,j); } cout << BFS(k) << endl; } }