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

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

原始题目链接：https://leetcode-cn.com/problems/longest-zigzag-path-in-a-binary-tree/

给你一棵以 root 为根的二叉树，二叉树中的交错路径定义如下：
选择二叉树中 任意 节点和一个方向（左或者右）。
如果前进方向为右，那么移动到当前节点的的右子节点，否则移动到它的左子节点。
改变前进方向：左变右或者右变左。
重复第二步和第三步，直到你在树中无法继续移动。
交错路径的长度定义为：访问过的节点数目 - 1（单个节点的路径长度为 0 ）。
请你返回给定树中最长 交错路径 的长度。

输入：root = [1,null,1,1,1,null,null,1,1,null,1,null,null,null,1,null,1]
输出：3 解释：蓝色节点为树中最长交错路径（右 -> 左 -> 右）。

输入：root = [1,1,1,null,1,null,null,1,1,null,1] 输出：4 解释：蓝色节点为树中最长交错路径（左
-> 右 -> 左 -> 右）。

解题思路：

使用深度优先搜索遍历，遍历的过程中每次更新最大交错路径值，交替变化左右两个反向，因为是树中任意节点开始，不满交替条件的节点，需要重置长度值，重置为1，因为从不满足的那个节点开始算起，已经走了长度为1的一条边。

代码实现：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def longestZigZag(self, root: TreeNode) -> int:
        # 初始化最大交错路径
        max_path = 0
        
        # 定义一个闭包函数：深度优先搜索dfs(root， dir, length)
        def dfs(root, direction, length):
            """
            root: 遍历到的当前的节点
            direction: 当前遍历的方向
            length: 目前最长的交错路径长度
            """
            # 递推结束条件：节点为空
            if not root:
                return
            nonlocal max_path

            # 每次都更新当前的最大值
            max_path = max(max_path, length)
            # 0表示左侧
            if direction == 0:
                # 当前方向向左，并且当前节点有左子树，即可以向左，长度自增1，下一步的方向向右
                dfs(root.left, 1, length + 1)
                # 当前方向是0应该向左，但是当前节点没有左子树，只能向右，长度重置为1，下一步的方向向左
                dfs(root.right, 0, 1)
            else:
                # 情况和0相反
                dfs(root.right, 0, length + 1)
                dfs(root.left, 1, 1)
        
        # 调用闭包函数
        dfs(root, 0, 0)
        dfs(root, 1, 0)

        # 返回结果
        return max_path

参考文献：
https://leetcode-cn.com/problems/longest-zigzag-path-in-a-binary-tree/solution/er-cha-shu-zhong-de-zui-chang-jiao-cuo-lu-jing-b-2/