二叉树之最长同值路径

太过爱你忘了你带给我的痛 2023-10-06 08:01 40阅读 0赞

## 二叉树之最长同值路径 ##

#### 问题： ####

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 1]

#### 思路: ####

###### 后序遍历 + 递归： ######

定义变量ans保存最长同值路径长度，初始为0

递归函数的功能为计算从p节点开始，向下出发最长的同值路径长度，递归函数的参数包含引用变量参数ans

先对p节点的左右子树根调用递归函数，得到从左右子树根出发的最长同值路径长度分别为left,right

若左右子树根的值与p节点的值相同，对应的最长同值路径长度 left 或 right 需加1（加上到根节点的一条边）。否则将 left 或 right 赋为0（从p节点开始，向下出发最长的同值路径长度为0）

若得到的 left+right>ans ，将ans 更新

最后函数返回，返回值为 left 与 right 中的最大值，即从p节点开始，向下出发最长的同值路径长度

遍历完成后的变量ans的值即为所求

#### 代码： ####

/**
     * Definition for a binary tree node.
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    /**
     * Definition for a binary tree node.
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution {
        
    public:
        int longestpathlen( TreeNode* p,int &ans )
        {
        
            if( p==NULL )
                return 0;
            int left=longestpathlen( p->left,ans );
            int right=longestpathlen( p->right,ans );
            if( p->left!=NULL && p->left->val==p->val )
                left++;
            else 
                left=0;
            if( p->right!=NULL && p->right->val==p->val )
                right++;
            else
                right=0;
            ans=max( ans,left+right );
            return max( left,right );
        }
        int longestUnivaluePath(TreeNode* root) {
        
            int ans=0;
            longestpathlen( root,ans );
            return ans;
        }
    };

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/2020050411043586.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20200504110451123.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw,size_16,color_FFFFFF,t_70