小红书面试代码

leetcode329

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

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

示例 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]。注意不允许在对角线方向上移动。

//不需要控制是否访问过 大小关系暗含了访问关系
//1.函数功能:计算该点为起点的最长序列的长度
int dfs(int x,int y,vector<vector<int>>&matrix,vector<vector<int>>&memo_len){
    //2.终止条件a.该节点已经访问过 直接返回
    if(memo_len[x][y]!=0)return memo_len[x][y];
    //2.终止条件b.该节点未访问过 递归返回结果
    vector<int>add_x={-1,1,0,0};
    vector<int>add_y={0,0,-1,1};
    int len=1;
    for(int i=0;i<4;i++){
        int new_x=x+add_x[i];
        int new_y=y+add_y[i];
        if(new_x>=0&&new_x<matrix.size()&&new_y>=0&&new_y<matrix[0].size()&&matrix[x][y]<matrix[new_x][new_y]){
            //3.等价关系式
            len=max(len,dfs(new_x,new_y,matrix,memo_len)+1);
        }
    }
    memo_len[x][y]=len;
    return memo_len[x][y];
}
class Solution {
public:
    int longestIncreasingPath(vector<vector<int>>& matrix) {
        if(matrix.size()==0)return 0;
        //记忆化搜索
        vector<vector<int>>memo_len(matrix.size(),vector<int>(matrix[0].size(),0));
        int max_len=1;
        for(int i=0;i<matrix.size();i++){
            for(int j=0;j<matrix[0].size();j++){
                max_len=max(max_len,dfs(i,j,matrix,memo_len));    
            }
        }
        return max_len;
    }
};