题目链接
题目描述
给定一个 m x n 整数矩阵 matrix ,找出其中 最长递增路径 的长度。
对于每个单元格,你可以往上,下,左,右四个方向移动。 不能 在 对角线 方向上移动或移动到 边界外(即不允许环绕)。
示例 1:
输入:matrix = [[9,9,4],[6,6,8],[2,1,1]]
输出:4
解释:最长递增路径为 [1, 2, 6, 9]。
示例 2:
输入:matrix = [[3,4,5],[3,2,6],[2,2,1]]
输出:4
解释:最长递增路径是 [3, 4, 5, 6]。注意不允许在对角线方向上移动。
示例 3:输入:matrix = [[1]]
输出:1
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 200
0 <= matrix[i][j] <= 231 - 1
解题思路
DFS+记忆化搜索
代码
Python
class Solution:
def longestIncreasingPath(self, matrix: list[list[int]]) -> int:
m, n = len(matrix), len(matrix[0])
res = [[0] * n for _ in range(m)] # 存储访问过的单元格走到最长递增路径
ret = 0
def isLegal(row: int, col: int) -> bool:
return 0 <= row < m and 0 <= col < n
def dfs(row: int, col: int) -> int:
if res[row][col] != 0: # res中已经存储了该单元的最长递增路径
return res[row][col]
ans = 1
cur = matrix[row][col]
if isLegal(row - 1, col) and matrix[row - 1][col] > cur: # up
ans = max(ans, dfs(row - 1, col) + 1)
if isLegal(row + 1, col) and matrix[row + 1][col] > cur: # down
ans = max(ans, dfs(row + 1, col) + 1)
if isLegal(row, col - 1) and matrix[row][col - 1] > cur: # left
ans = max(ans, dfs(row, col - 1) + 1)
if isLegal(row, col + 1) and matrix[row][col + 1] > cur: # right
ans = max(ans, dfs(row, col + 1) + 1)
res[row][col] = ans
return ans
for i in range(m):
for j in range(n):
ret = max(ret, dfs(i, j))
return ret