题目浅析

  • 想查看原题可以点击题目链接。

  • 简单地说，就是给一个二叉树，现在要在其中找到一条最长的路径，路径中要求值相同。

思路分享

代码解答（强烈建议自行解答后再看）

  • 参考题解
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# 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:
        ans = 0
        def dfs(root: Optional[TreeNode]) -> int:
            if not root:
                return -1 
            left = dfs(root.left) + 1
            right = dfs(root.right) + 1
            
            if root.left and root.val != root.left.val:
                left = 0
            if root.right and root.val != root.right.val:
                right = 0

            cur = left + right
            nonlocal ans 
            if ans < cur: 
                ans = cur 
            
            return max(left, right)
        
        dfs(root)
        return ans