329. 矩阵中的最长递增路径(C++)---记忆化深度优先搜索 解题

题目详情

给定一个整数矩阵，找出最长递增路径的长度。

对于每个单元格，你可以往上，下，左，右四个方向移动。 你不能在对角线方向上移动或移动到边界外（即不允许环绕）。

示例 1:
输入: nums = 

[
  [9,9,4],
  [6,6,8],
  [2,1,1]
] 
输出: 4 
解释: 最长递增路径为 [1, 2, 6, 9]。

示例 2:
输入: nums = 

[
  [3,4,5],
  [3,2,6],
  [2,2,1]
] 
输出: 4 
解释: 最长递增路径是 [3, 4, 5, 6]。注意不允许在对角线方向上移动。

 

——题目难度:困难


 

---

分析

如果只是使用朴素深度优先搜索，因为同一个单元格会被访问多次，每次访问都需要重新计算，所以进行了大量的重复计算。
由于同一个单元格对应的最长递增路径的长度是固定不变的，因此可以使用记忆化的方法进行优化。用矩阵 memo 作为缓存(记忆)矩阵，将已经计算过的单元格的结果存储到缓存矩阵中。

使用记忆化深度优先搜索，当访问到一个单元格(i, j)时，如果 memo[i][j] ≠ 0，说明该单元格已经被访问过，则直接从缓存中读取结果；如果 memo[i][j] = 0，说明单元格尚未访问过，则可以对该单元格进行搜索，并将计算得到的结果值存入 memo 矩阵中。

遍历完矩阵中的所有单元格后，即可得到矩阵中的最长递增路径的长度。


-解题代码

class Solution {
public:
	int dirs[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
	int rows, columns;
	
	int dfs(vector<vector<int>>& matrix, int row, int column, vector<vector<int>>& memo) {
		if (memo[row][column]) return memo[row][column];
		
		memo[row][column]++;
		for(int i = 0; i < 4; i++) {
			int newRow = row + dirs[i][0], newColumn = column + dirs[i][1];
			if (newRow >= 0 && newRow < rows && newColumn >= 0 && newColumn < columns && matrix[newRow][newColumn] > matrix[row][column]) {
				memo[row][column] = max(memo[row][column], 1 + dfs(matrix, newRow, newColumn, memo));
			}
		}
		
		return memo[row][column];
	}
	
    int longestIncreasingPath(vector<vector<int>>& matrix) {
    	/*这里有点细节，当matrix为空时，matrix.size() == 0为真，则避免了下一个matrix[0].size()的计算*/
		/*因为如果此时计算matrix.size() == 0 则会导致越界*/ 
    	if (matrix.size() == 0 || matrix[0].size() == 0) return 0;
    	
		rows = matrix.size();
		columns = matrix[0].size();
		//memo(memory)为缓存矩阵 ——负责记录 已经计算过的单元格的结果 
		vector<vector<int>> memo(rows, vector<int>(columns, 0)); 
		int ans = 0;
		for(int i = 0; i < rows; i++) {
			for(int j = 0; j < columns; j++) {
				ans = max(ans, dfs(matrix, i, j, memo));
			}
		}
		
		return ans;
    }
};