LeetCode 687 最长同值路径[dfs] HERODING的LeetCode之路

Posted 2022-09-05 HERODING23

解题思路：
一道二叉树深度搜索的变型，在dfs的模板上稍作修改即可，首先定义全局变量maxlen用于更新最长的路径，用先序的思想遍历二叉树，一直往下，返回左子树和右子树所在子树的最大路径长度，并更新全局最大长度，到达最底部的时候判断当前节点值和父节点的值是否想相同，相同返回左右节点的最大值+1,（这是因为必须在一条线上，所以更新到父节点的时候要舍弃短的那条线），代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode 
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) 
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) 
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) 
 * ;
 */
class Solution 
private:
    int maxLen = 0;
public:
    int longestUnivaluePath(TreeNode* root) 
        if(root == nullptr) return 0;
        dfs(root, root->val);
        return maxLen;
    

    int dfs(TreeNode* root, int val) 
        if(root == nullptr) 
            return 0;
        
        int left = dfs(root->left, root->val);
        int right = dfs(root->right, root->val);
        maxLen = max(maxLen, left + right);
        if(root->val == val) 
            return max(left, right) + 1;
        
        return 0;        
    
;