【Leetcode】980. Unique Paths III

题目地址：

https://leetcode.com/problems/unique-paths-iii/

给定一个$m$行$n$列的二维矩阵，$0$代表空地，$-1$代表障碍物，$1$代表起点，$2$代表终点。问从起点开始走过所有空地到达终点的路径方案数。

可以用DFS + 状态压缩。用一个整型数字的二进制位代表哪些位置已经访问过。代码如下：

class Solution {
 public:
  int uniquePathsIII(vector<vector<int>>& g) {
    int sx, sy, cnt = 0;
    for (int i = 0; i < g.size(); i++)
      for (int j = 0; j < g[0].size(); j++)
        if (g[i][j] == 1) sx = i, sy = j, cnt++;
        else if (!g[i][j]) cnt++;

    int res = 0;
    dfs(sx, sy, 0, cnt, g, res);
    return res;
  }

  void dfs(int x, int y, int vis, int cnt, vector<vector<int>>& g, int& res) {
    if (g[x][y] == 2) {
      if (!cnt) res++;
      return;
    }

    vis |= 1 << x * g[0].size() + y;
    cnt--;
    static int d[] = {-1, 0, 1, 0, -1};
    for (int i = 0; i < 4; i++) {
      int nx = x + d[i], ny = y + d[i + 1];
      if (0 <= nx && nx < g.size() && 0 <= ny && ny < g[0].size()) {
        int nidx = nx * g[0].size() + ny;
        if (!~g[nx][ny] || (vis >> nidx & 1)) continue;
        dfs(nx, ny, vis, cnt, g, res);
      }
    }
  }
};

时间复杂度是指数级，空间$O(mn)$。