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

发布时间 2023-05-20 20:26:21作者: Owlwu

给你一棵以 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] 之间。

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/longest-zigzag-path-in-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


 

class Solution {
    // 记录最大值。
    int max = 0;

    public int longestZigZag (TreeNode root) {
        // 由于左右子树都会考虑到,所以此处是 false 还是 true 都没有影响。
        dfs(root, true);
        return max;
    }
// @isLeft: 是否是从左边来的。(如果是,则向右拐,否则向左拐)。                               
    private int dfs (TreeNode root, boolean isLeft) {
        if (root == null) {
            return 0;
        }
        // 左子树的最大值
        int r = dfs(root.right, false);
        // 右子树的最大值
        int l = dfs(root.left, true);
        int big = Math.max(l, r);
        // 更新最大值
        if (big > max) {
            max = big;
        }
        // 之前是左拐的,用右子树的最大值,之前是右拐的,用左子树的最大值。
        if (isLeft) {
            return r + 1;
        } else {
            return l + 1;
        }
    }
}