二叉树之最长同值路径
问题:
思路:
后序遍历 + 递归:
定义变量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;
}
};