给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。
注意:两个节点之间的路径长度由它们之间的边数表示。、
——————————————
解题思路:使用递归算法,使用一个递归函数计算与当前节点值相同的左子树的连续同值个数或者右子树的连续同值个数。
另外定义一个变量记录当前结点的最长同值路径。使用Python计算如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def __init__(self):
self.ans = 0
def longestUnivaluePath(self, root: TreeNode) -> int:
def longarrow(root): # 计算以当前节点为起点的最长同值左连续路径或者最长同值右连续路径
if not root:
return 0
leftlength = longarrow(root.left) # 获得左子树的最长连续同值个数
rightlength = longarrow(root.right) # 获得右子树的最长连续同值个数
leftarrow = 0
rightarrow = 0
if root.left and root.val==root.left.val # 如果左子树的值和根节点的值相同:
leftarrow = leftlength + 1
if root.right and root.val == root.right.val: # 如果右节点的值和根节点的值相同
rightarrow = rightlength + 1
self.ans = max(self.ans,leftarrow + rightarrow) # 更新每个节点的最长连续同值个数
return max(leftarrow,rightarrow) # 返回左子树或者右子树的最长同值节点个数
longarrow(root)
return self.ans