← 题库
中等深度优先搜索动态规划

PROBLEM / 研究生阶段

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

给你一棵以 root 为根的二叉树,二叉树中的交错路径定义如下:

  • 选择二叉树中任意 节点和一个方向(左或者右)。
  • 如果前进方向为右,那么移动到当前节点的的右子节点,否则移动到它的左子节点。
  • 改变前进方向:左变右或者右变左。
  • 重复第二步和第三步,直到你在树中无法继续移动。

交错路径的长度定义为:访问过的节点数目 - 1(单个节点的路径长度为 0 )。 请你返回给定树中最长 交错路径 的长度。

示例 1:

图片
图片

输入: 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] 之间。

参考解法

下面保留的是原始练习仓库中的个人作答,可在右侧工作台中独立重写,再按需揭示对照。

python

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def longestZigZag(self, root):
        self.max_len = 0
        def dfs(node):
            if not node:
                return (-1, -1)
            # 递归获取左右子树的结果
            _, left_right = dfs(node.left)
            right_left, _ = dfs(node.right)
            # 当前节点向左:依赖左子节点的向右长度 + 1
            curr_left = left_right + 1
            # 当前节点向右:依赖右子节点的向左长度 + 1
            curr_right = right_left + 1
            
            self.max_len = max(self.max_len, curr_left, curr_right)
            return (curr_left, curr_right)
        
        dfs(root)
        return self.max_len
        """
        :type root: Optional[TreeNode]
        :rtype: int
        """
这道题暂未附参考解法,编辑器草稿仍会自动保存在本机。
YOUR SOLUTION1372. 二叉树中的最长交错路径