Leetcode 329.矩阵中的最长递增路径
1 题目描述(Leetcode题目链接)
给定一个整数矩阵,找出最长递增路径的长度。
对于每个单元格,你可以往上,下,左,右四个方向移动。 你不能在对角线方向上移动或移动到边界外(即不允许环绕)。
输入: nums =
[
[9,9,4],
[6,6,8],
[2,1,1]
]
输出: 4
解释: 最长递增路径为 [1, 2, 6, 9]。
输入: nums =
[
[3,4,5],
[3,2,6],
[2,2,1]
]
输出: 4
解释: 最长递增路径是 [3, 4, 5, 6]。注意不允许在对角线方向上移动。
2 题解
最直观的想法就是对每一个点进行深度优先搜索,最后取最大值,但会超时,因为会多出很多冗余的计算。因此需要做一些优化,即记录每一个点的结果,下一次再搜索到这个点的时候就不用再计算了。
令 d p [ i ] [ j ] dp[i][j] dp[i][j]表示以 m a t r i x [ i ] [ j ] matrix[i][j] matrix[i][j]为起点的最长递增路径,初始化为全 1 1 1,然后对每个点进行深度优先搜索,同时更新 d p [ i ] [ j ] dp[i][j] dp[i][j],最后返回 d p dp dp中最大的值。
class Solution:
def longestIncreasingPath(self, matrix: List[List[int]]) -> int:
if not matrix:
return 0
m, n = len(matrix), len(matrix[0])
res = 1
dp = [([1]*n) for _ in range(m)]
def dfs(i, j):
if dp[i][j] != 1:
return dp[i][j]
for di in [(0, 1), (0, -1), (1, 0), (-1, 0)]:
x, y = i + di[0], j + di[1]
if x < 0 or y < 0 or x >= m or y >= n:
continue
if matrix[i][j] < matrix[x][y]:
dp[i][j] = max(dp[i][j], 1 + dfs(x, y))
return dp[i][j]
for i in range(m):
for j in range(n):
dfs(i, j)
res = max(res, dp[i][j])
return res