题目描述:
给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。
注意:两个节点之间的路径长度由它们之间的边数表示。
示例 1:
输入:
5
/ \
4 5
/ \ \
1 1 5
输出:
2
示例 2:
输入:
1
/ \
4 5
/ \ \
4 4 5
输出:
2
注意: 给定的二叉树不超过10000个结点。 树的高度不超过1000。
方法1:递归
主要思路:
(1)对题目进行分析,求最大的相同数值的路径,对于当前结点作为根节点的话,若是当前结点的值和左子树的值相等,则可以将左子树到达此处时其相同的路径长度值加一,否则,就是0,因为已经值不相同了;
(2)同样,对于右子树,若是当前结点的值和右子树的值相等,则可以将右子树到达此处时其相同的路径长度值加一,否则,就是0,因为已经值也不相同了;
(3)则对于当前结点作为根节点,则此时将其作为路径的中间值,既左子树的路径值加上右子树的路径值之和,为一个可能的最长相同路径值,进行判断保留;
(4)则若是将当前结点作为路径中的一个子树的部分,则需要在左右子树中选择一个较大的值,作为其当前路径的值,作为函数的返回值,接着向上回退,找可能的更大的值;
/**
* 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 helper(TreeNode* root,int& count){
//终止条件
if(root==NULL)
return 0;
//左右递归
int left_side=helper(root->left,count);
int right_side=helper(root->right,count);
//统计左右子树的值能不能和当前结点的值结合
int left_path=0;
int right_path=0;
if(root->left&&root->val==root->left->val){//若是可以,则在左子树的路径值基础上加一
left_path=left_side+1;
}
if(root->right&&root->val==root->right->val){//若是可以,则在右子树的路径值基础上加一
right_path=right_side+1;
}
count=max(count,right_path+left_path);//保留可能的最大值,既将当前根节点作为路径的中间值,不向上回退
return max(left_path,right_path);//将当前结点作为子树路径的一部分,在左右子树中选个较大的,接着向上回退
}
int longestUnivaluePath(TreeNode* root) {
int count=0;//保存最长的同值路径的长度
helper(root,count);
return count;
}
};