【洛谷】P2802 回家

题目地址：

https://www.luogu.com.cn/problem/P2802

题目描述：
小H在一个划分成了$n\times m$个方格的长方形封锁线上。 每次他能向上下左右四个方向移动一格（当然小H不可以静止不动）， 但不能离开封锁线，否则就被打死了。 刚开始时他有满血$6$点，每移动一格他要消耗$1$点血量。一旦小H的血量降到$0$， 他将死去。 他可以沿路通过拾取鼠标（什么鬼。。。）来补满血量。只要他走到有鼠标的格子，他不需要任何时间即可拾取。格子上的鼠标可以瞬间补满，所以每次经过这个格子都有鼠标。就算到了某个有鼠标的格子才死去， 他也不能通过拾取鼠标补满HP。 即使在家门口死去， 他也不能算完成任务回到家中。地图上有五种格子：
0：障碍物。
1：空地， 小 H 可以自由行走。
2：小 H 出发点， 也是一片空地。
3：小 H 的家。
4：有鼠标在上面的空地。
小 H 能否安全回家？如果能， 最短需要多长时间呢？

输入格式：
第一行两个整数$n,m$， 表示地图的大小为$n\times m$。
下面$n$行， 每行$m$个数字来描述地图。

输出格式：
一行， 若小H不能回家， 输出-1，否则输出他回家所需最短时间。

以元组$(x,y,e)$为状态，$(x,y)$是坐标，$e$是血量。初始状态是$(sx,sy,6)$（$(sx,sy)$是初始位置），目标状态是$(hx,hy,.>0)$（$(hx,hy)$是目标位置，只需要到达目标位置的时候血量大于$0$即可）。BFS即可，每一步不能出界，不能走到障碍物，不能走到之前走过的状态，也不能走到血量变为$0$的状态。每一层BFS记录步数，如果能走到家就返回步数即可。代码如下：

#include <iostream>
#include <queue>
using namespace std;

const int N = 15;
int m, n;
int a[N][N];
int sx, sy, hx, hy;
struct Node {
  int x, y;
  int e;
};
queue<Node> q;
bool vis[N][N][10];

int bfs() {
  Node node = {sx, sy, 6};
  vis[sx][sy][6] = true;
  q.push(node);
  static int d[] = {-1, 0, 1, 0, -1};
  int res = 0;
  while (q.size()) {
    res++;
    for (int _ = q.size(); _; _--) {
      auto t = q.front(); q.pop();
      int x = t.x, y = t.y, e = t.e;
      // 如果当前血量为1，无论走到哪儿都会死去，直接略过
      if (e == 1) continue;
      for (int i = 0; i < 4; i++) {
        int nx = x + d[i], ny = y + d[i + 1];
        // 不出界
        if (1 <= nx && nx <= m && 1 <= ny && ny <= n) {
          // 走到家了，返回步数
          if (nx == hx && ny == hy) return res;
          // 不能走障碍物
          if (!a[nx][ny]) continue;
          // 计算一下走到(nx, ny)位置的状态
          Node ne = {nx, ny, a[nx][ny] == 4 ? 6 : e - 1};
          // 之前走过的位置不走
          if (vis[nx][ny][ne.e]) continue;
          vis[nx][ny][ne.e] = true;
          q.push(ne);
        }
      }
    }
  }

  return -1;
}

int main() {
  scanf("%d%d", &m, &n);
  for (int i = 1; i <= m; i++)
    for (int j = 1; j <= n; j++) {
      int x;
      scanf("%d", &x);
      a[i][j] = x;
      if (x == 2)
        sx = i, sy = j;
      else if (x == 3)
        hx = i, hy = j;
    }

  printf("%d\n", bfs());
}

时空复杂度$O(mn)$。