二叉树之最长同值路径

二叉树之最长同值路径

问题：

思路:

后序遍历 + 递归：

定义变量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;
    }
};