题目:
给定一个二叉树的 root ,返回 最长的路径的长度 ,这个路径中的 每个节点具有相同值 。 这条路径可以经过也可以不经过根节点。
两个节点之间的路径长度 由它们之间的边数表示。
提示:
- 树的节点数的范围是 [0, 104]
- -1000 <= Node.val <= 1000
- 树的深度将不超过 1000
解答:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def longestUnivaluePath(self, root: Optional[TreeNode]) -> int:
self.res=0
#返回从根节点出发的最长同值路径
def dfs(root):
if not root:
return 0
L=dfs(root.left)
R=dfs(root.right)
L=L+1 if root.left and root.val==root.left.val else 0
R=R+1 if root.right and root.val==root.right.val else 0
self.res=max(self.res,L+R)
return max(L,R)
dfs(root)
return self.res