1372. 二叉树中的最长交错路径

Difficulty
Medium
Tags
二叉树
URL
https://leetcode-cn.com/problems/longest-zigzag-path-in-a-binary-tree/
Star
给你一棵以 root 为根的二叉树，二叉树中的交错路径定义如下：
  • 选择二叉树中 任意 节点和一个方向（左或者右）。
  • 如果前进方向为右，那么移动到当前节点的的右子节点，否则移动到它的左子节点。
  • 改变前进方向：左变右或者右变左。
  • 重复第二步和第三步，直到你在树中无法继续移动。
交错路径的长度定义为：访问过的节点数目 - 1（单个节点的路径长度为 0 ）。
请你返回给定树中最长 交错路径 的长度。
示例 1：
notion imagenotion image
输入：root = [1,null,1,1,1,null,null,1,1,null,1,null,null,null,1,null,1]
输出：3
解释：蓝色节点为树中最长交错路径（右 -> 左 -> 右）。
示例 2：
输入：root = [1,1,1,null,1,null,null,1,1,null,1]
输出：4
解释：蓝色节点为树中最长交错路径（左 -> 右 -> 左 -> 右）。
示例 3：
输入：root = [1]
输出：0
提示：
  • 每棵树最多有 50000 个节点。
  • 每个节点的值在 [1, 100] 之间。

法 1 信息传递

思路
采用后序遍历，从子节点往上传递信息，传递的信息包括子节点的左右两边交错路径的长度。
这样当前节点就有了左右子节点的具体信息，那么当前节点就可以继续往上传递了。
notion imagenotion image
题解
# 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 longestZigZag(self, root: TreeNode) -> int:
        self.res = 0
        self.helper(root)
        return self.res-1 if self.res > 0 else 0

    def helper(self, root):
        if root is None:
            # [leftsum, rightsum]
            # 左边的交错路径上的节点数，右边的。。。
            return (0, 0)  

        l = self.helper(root.left)
        r = self.helper(root.right)
        # 当前节点的 左右两边的交错路径节点数目
        temp = [l[1]+1, r[0]+1]
        self.res = max(self.res, max(temp))

        return temp