【980. 不同路径 III】

来源:力扣(LeetCode)

描述:

在二维网格 grid 上,有 4 种类型的方格:

  • 1 表示起始方格。且只有一个起始方格。
  • 2 表示结束方格,且只有一个结束方格。
  • 0 表示我们可以走过的空方格。
  • -1 表示我们无法跨越的障碍。

返回在四个方向(上、下、左、右)上行走时,从起始方格到结束方格的不同路径的数目。

每一个无障碍方格都要通过一次,但是一条路径中不能重复通过同一个方格。

示例 1:

输入:[[1,0,0,0],[0,0,0,0],[0,0,2,-1]]
输出:2
解释:我们有以下两条路径:
1. (0,0),(0,1),(0,2),(0,3),(1,3),(1,2),(1,1),(1,0),(2,0),(2,1),(2,2)
2. (0,0),(1,0),(2,0),(2,1),(1,1),(0,1),(0,2),(0,3),(1,3),(1,2),(2,2)

示例 2:

输入:[[1,0,0,0],[0,0,0,0],[0,0,0,2]]
输出:4
解释:我们有以下四条路径: 
1. (0,0),(0,1),(0,2),(0,3),(1,3),(1,2),(1,1),(1,0),(2,0),(2,1),(2,2),(2,3)
2. (0,0),(0,1),(1,1),(1,0),(2,0),(2,1),(2,2),(1,2),(0,2),(0,3),(1,3),(2,3)
3. (0,0),(1,0),(2,0),(2,1),(2,2),(1,2),(1,1),(0,1),(0,2),(0,3),(1,3),(2,3)
4. (0,0),(1,0),(2,0),(2,1),(1,1),(0,1),(0,2),(0,3),(1,3),(1,2),(2,2),(2,3)

示例 3:

输入:[[0,1],[2,0]]
输出:0
解释:
没有一条路能完全穿过每一个空的方格一次。
请注意,起始和结束方格可以位于网格中的任意位置。

提示:

  • 1 <= grid.length * grid[0].length <= 20

方法一:回溯

思路

按照要求,假设矩阵中有 n 个 0,那么一条合格的路径,是长度为 (n+1),由 1 起始,结束于 2,不经过 −1,且每个点只经过一次的路径。要求出所有的合格的路径,可以采用回溯法,定义函数 dfs,表示当前 grid 状态下,从点 (i, j) 出发,还要经过 n 个点,走到终点的路径条数。到达一个点时,如果当前的点为终点,且已经经过了 (n+1) 个点,那么就构成了一条合格的路径,否则就不构成。如果当前的点不为终点,则将当前的点标记为 −1,表示这条路径以后不能再经过这个点,然后继续在这个点往四个方向扩展,如果不超过边界且下一个点的值为 0 或者 2,则表示这条路径可以继续扩展。探测完四个方向后,需要将当前的点的值改为原来的值。将四个方向的合格路径求和,即为当前状态下合格路径的条数。最终需要返回的是,grid 在初始状态下,从起点出发,需要经过 (n+1) 个点的路径条数。

代码:

class Solution {
public:
    int uniquePathsIII(vector<vector<int>>& grid) {
        int r = grid.size(), c = grid[0].size();
        int si = 0, sj = 0, n = 0;
        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {
                if (grid[i][j] == 0) {
                    n++;
                 } else if (grid[i][j] == 1) {
                    n++;
                    si = i;
                    sj = j;
                 }
            }
        }

        function<int(int, int, int)> dfs = [&](int i, int j, int n) -> int {
            if (grid[i][j] == 2) {
                if (n == 0) {
                    return 1;
                }
                return 0;
            }

            int t = grid[i][j], res = 0;
            grid[i][j] = -1;
            vector<array<int, 2>> dir({{-1, 0}, {1, 0}, {0, -1}, {0, 1}});
            for (auto &[di, dj] : dir) {
                int ni = i + di;
                int nj = j + dj;
                if (ni >= 0 && ni < r && nj >= 0 && nj < c && \
                   (grid[ni][nj] == 0 || grid[ni][nj] == 2)) {
                    res += dfs(ni, nj, n - 1);
                }
            }
            grid[i][j] = t;
            return res;
        };
        return dfs(si, sj, n);
    }
};

执行用时:8 ms, 在所有 C++ 提交中击败了39.82%的用户
内存消耗:8 MB, 在所有 C++ 提交中击败了26.99%的用户
复杂度分析

  • 时间复杂度:O(4r × c),其中 r 和 c 分别是 grid 的行数和列数。
  • 空间复杂度:O(r × c),是回溯的深度。

方法二:记忆化搜索 + 状态压缩

思路

方法一的回溯函数,即使在 i, j, n 都相同的情况下,函数的返回值也会不同,因为路径经过的点不同,导致当前情况下 grid 的状态也不同。因此,我们可以将grid 的状态放入函数的输入参数,从而用记忆化搜索来降低时间复杂度。

用一个二进制数字 st 来表示路径还未经过的点(初始状态下为所有值为 0 的点和终点),点的坐标需要和二进制数字的位一一对应。定义函数 dp,输入参数为当前坐标 i, j 和为经过的点的二进制集合 st,返回值即为从点 (i, j) 出发,经过 st 代表的点的集合,最终到达终点的路径的条数。如果当前点为终点且剩下没有未经过的点,那么当前的返回值即为 1,否则为 0。如果当前的点不为终点,则需要探索四个方向,如果接下来的点在边界内且还未经过(用按位和操作判断),则需要走到那个点并且将那个点的坐标从未经过的点的集合中去掉(用按位异或操作)。将四个方向的合格路径求和,即为当前状态下合格路径的条数。最终需要返回的是,从起点出发,为经过的点的集合为所有值为 0 的点和终点的路径条数。函数调用过程中采用了记忆化搜索,每个状态最多只会被计算一次。

代码:

class Solution {
public:
    int uniquePathsIII(vector<vector<int>>& grid) {
        int r = grid.size(), c = grid[0].size();
        int si = 0, sj = 0, st = 0;
        unordered_map<int, int> memo;
        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {
                if (grid[i][j] == 0 || grid[i][j] == 2) {
                    st |= (1 << (i * c + j));
                } else if (grid[i][j] == 1) {
                    si = i, sj = j;
                }
            }
        }

        function<int(int ,int, int)> dp = [&](int i, int j, int st) -> int {
            if (grid[i][j] == 2) {
                if (st == 0) {
                    return 1;
                }
                return 0;
            }
            int key = ((i * c + j) << (r * c)) + st;
            if (!memo.count(key)) {
                int res = 0;
                vector<array<int, 2>> dir({{-1, 0}, {1, 0}, {0, -1}, {0, 1}});
                for (auto &[di, dj] : dir) {
                    int ni = i + di, nj = j + dj;
                    if (ni >= 0 && ni < r && nj >= 0 && nj < c && (st & (1 << (ni * c + nj))) > 0) {
                        res += dp(ni, nj, st ^ (1 << (ni * c + nj)));
                    }
                }
                memo[key] = res;
            }
            return memo[key];
        };
        return dp(si, sj, st);
    }
};

执行用时:20 ms, 在所有 C++ 提交中击败了26.55%的用户
内存消耗:9.2 MB, 在所有 C++ 提交中击败了26.55%的用户
复杂度分析

  • 时间复杂度:O(r × c × 2r×c),其中 r 和 c 分别是 grid 的行数和列数。O(r × c × 2r×c) 是状态数,每个状态只会被计算一次,计算一个状态的时间复杂度为 O(1)。
  • 空间复杂度:O(r × c × 2r×c),是状态数。
    author:LeetCode-Solution
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

千北@

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值