Уникальные пути III

Вам дан целочисленный массив m x n grid, где grid[i][j] может быть:

• 1, представляющий начальную клетку. Существует ровно один начальный квадрат.
• 2, представляющий конечный квадрат. Существует ровно одна конечная клетка.
• 0 представляет пустые квадраты, по которым можно ходить.
• -1, представляющие препятствия, через которые мы не можем пройти.

Верните количество 4-направленных ходов от начальной до конечной клетки, которые проходят через каждую клетку без препятствий ровно один раз.

Пример 1:

Вход: grid = [[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:

Вход: grid = [[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:

Вход: grid = [[0,1],[2,0]].
Выход: 0
Объяснение: Не существует пути, который проходит через каждый пустой квадрат ровно один раз.
Обратите внимание, что начальная и конечная клетка может находиться в любом месте сетки.

Ограничения:

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 20
• 1 <= m * n <= 20
• -1 <= grid[i][j] <= 2
• Существует ровно одна начальная и одна конечная ячейка.

РЕШЕНИЕ:

class Solution:
    def getNeighbors(self, start, m, n):
        a, b = start
        neighbors = []
        if a > 0:
            neighbors.append((a - 1, b))
        if a < m - 1:
            neighbors.append((a + 1, b))
        if b > 0:
            neighbors.append((a, b - 1))
        if b < n - 1:
            neighbors.append((a, b + 1))
        return neighbors

    def unipaths(self, start, grid, visited, m, n, numEmpty):
        currval = grid[start[0]][start[1]]
        ways = 0
        if currval == -1:
            return ways
        if currval == 2:
            if len(visited) == numEmpty + 2:
                return ways + 1
            return ways
        neighbors = self.getNeighbors(start, m, n)
        for neighbor in neighbors:
            if grid[neighbor[0]][neighbor[1]] != -1 and neighbor not in visited:
                ways += self.unipaths(neighbor, grid, visited.union({neighbor}), m, n, numEmpty)
        return ways

    def uniquePathsIII(self, grid: List[List[int]]) -> int:
        m = len(grid)
        n = len(grid[0])
        numEmpty = 0
        start = None
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 0:
                    numEmpty += 1
                elif grid[i][j] == 1:
                    start = (i, j)
        return self.unipaths(start, grid, {start}, m, n, numEmpty)