687. 最长同值路径
1.题目描述
给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。
注意:两个节点之间的路径长度由它们之间的边数表示。
示例 1:
输入:
输出:
示例 2:
输入:
输出:
注意: 给定的二叉树不超过10000个结点。 树的高度不超过1000。
2.思路
深度优先遍历二叉树,如果当前节点的左子节点等于当前节点,则左路径leftpath = left + 1,如果当前节点的右子节点等于当前节点,则右路径rightpath = right + 1,遍历过程中取左右路径之和的最大值。
3.代码
/**
* 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 longestUnivaluePath(TreeNode* root) {
int path = 0;
dfs(root,path);
return path;
}
int dfs(TreeNode* root, int& path){
if(root == NULL){
return 0;
}
int left = dfs(root->left,path);
int right = dfs(root->right,path);
int leftpath = root->left != NULL && root->val == root->left->val ? left + 1 : 0;
int rightpath = root->right != NULL && root->val == root->right->val ? right + 1 : 0;
path = max(path,leftpath+rightpath);
return max(leftpath,rightpath);
}
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)