Leetcode 980.不同路径 III（Unique Paths III）

Leetcode 980.不同路径 III

1 题目描述（Leetcode题目链接）

  在二维网格 grid 上，有 4 种类型的方格：

• 1 表示起始方格。且只有一个起始方格。
• 2 表示结束方格，且只有一个结束方格。
• 0 表示我们可以走过的空方格。
• -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)

输入：[[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)

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

提示：1 <= grid.length * grid[0].length <= 20

2 题解

  首先遍历网格，统计0的数量并记录起点的位置，然后从起点开始深度优先搜索就可以了。

class Solution:
    def uniquePathsIII(self, grid: List[List[int]]) -> int:
        begin = [0, 0]
        zeros = 1
        m, n = len(grid), len(grid[0])
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 1:
                    begin = [i, j]
                elif grid[i][j] == 0:
                    zeros += 1
        
        def dfs(i, j, zero_num):
            if i < 0 or i >= m or j < 0 or j >= n or grid[i][j] == -1:
                return 0
            if grid[i][j] == 2:
                if zero_num == 0:
                    return 1
                else:
                    return 0
            grid[i][j] = -1
            u = dfs(i-1, j, zero_num-1)
            d = dfs(i+1, j, zero_num-1)
            r = dfs(i, j+1, zero_num-1)
            l = dfs(i, j-1, zero_num-1)
            grid[i][j] = 0
            return u + d + r + l

        return dfs(begin[0], begin[1], zeros)