思想:还是求最大路径长度,所以还是套后续遍历的模板,先分别求左右子树中符合要求的最大长度,然后再判断root与root.left,root.right是否相等,若相等则过root可以构成成一条路径其长度 = leftLen+rightLen 。
时间复杂度O(n),空间复杂度O(logn)
class Solution {
int res;
public int helper(TreeNode root){
if(root==null)return 0;
int leftStaticLen = helper(root.left);//注意这里不能写成leftStaticLen = (root.left!=null&&root.left.val==root.val)?helper(root.left):0;因为这样当不满足条件时就没法取对子树进行递归统计了
int rightStaticLen = helper(root.right);
int leftLen = (root.left!=null&&root.left.val==root.val)?leftStaticLen:0;
int rightLen = (root.right!=null&&root.right.val==root.val)?rightStaticLen:0;
res = Math.max(leftLen+rightLen,res);
return Math.max(leftLen,rightLen)+1;
}
public int longestUnivaluePath(TreeNode root) {
if(root==null||root.left==null&&root.right==null)return 0;
helper(root);
return res;
}
}