Leetcode687.Longest Univalue Path最长同值路径

时间:2021-01-04 15:51:51

给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。

注意:两个节点之间的路径长度由它们之间的边数表示。

示例 1:

输入:

5 / \ 4 5 / \ \ 1 1 5

输出:

2

示例 2:

输入:

1 / \ 4 5 / \ \ 4 4 5

输出:

2

注意: 给定的二叉树不超过10000个结点。 树的高度不超过1000。

struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
}; class Solution {
public:
int longestUnivaluePath(TreeNode* root) {
if(root == NULL)
return 0;
int temp = max(Find1(root, root ->val), Find2(root, root ->val));
if(temp >= 2)
temp = temp - 1;
else
temp = 0;
return max(temp, max(longestUnivaluePath(root ->left), longestUnivaluePath(root ->right)));
} int Find1(TreeNode* root, int val)
{
if(root == NULL)
return 0;
if(root ->val == val)
return 1 + Find2(root ->left, val) + Find2(root ->right, val);
return 0;
} int Find2(TreeNode * root, int val)
{
if(root == NULL)
return 0;
if(root ->val == val)
return 1 + max(Find2(root ->left, val), Find2(root ->right, val));
return 0;
}
};