题目描述
给定一个二叉树的 root ,返回 最长的路径的长度 ,这个路径中的 每个节点具有相同值 。 这条路径可以经过也可以不经过根节点。
两个节点之间的路径长度 由它们之间的边数表示。
示例
- 示例 1:
输入:root = [5,4,5,1,1,5]
输出:2
- 示例 2:
输入:root = [1,4,5,4,4,5]
输出:2
提示
- 树的深度将不超过 1000
- -1000 <= Node.val <= 1000
- 树的节点数的范围是 [0, 104]
解题思路
- 深度优先算法解题,注意父节点与子节点值不同时,左右路径长度的更新
- 设置全局变量,记录路径最大值
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int temp;
int my_func(TreeNode *root){
if(root == NULL){
return 0;
}
int left, right, left_, right_;
left = my_func(root->left);
if(root->left != NULL && root->val == root->left->val){
left_ = left + 1;
}else{
left_ = 0;
}
right = my_func(root->right);
if(root->right != NULL && root->val == root->right->val){
right_ = right + 1;
}else{
right_ = 0;
}
temp = temp > (left_ + right_) ? temp : (left_ + right_);
return left_ > right_ ? left_ : right_;
}
int longestUnivaluePath(TreeNode* root) {
temp = 0;
my_func(root);
return temp;
}
};
提交结果
总结
- 深度优先遍历