LeetCode刷题系列 -- 1372. 二叉树中的最长交错路径

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

  • 选择二叉树中 任意 节点和一个方向(左或者右)。

  • 如果前进方向为右,那么移动到当前节点的的右子节点,否则移动到它的左子节点。

  • 改变前进方向:左变右或者右变左。

  • 重复第二步和第三步,直到你在树中无法继续移动。

交错路径的长度定义为:访问过的节点数目 - 1(单个节点的路径长度为 0 )。

请你返回给定树中最长 交错路径 的长度。

示例 1:

LeetCode刷题系列 -- 1372. 二叉树中的最长交错路径_第1张图片

输入:root = [1,null,1,1,1,null,null,1,1,null,1,null,null,null,1,null,1]

输出:3

解释:蓝色节点为树中最长交错路径(右 -> 左 -> 右)。

示例 2:

LeetCode刷题系列 -- 1372. 二叉树中的最长交错路径_第2张图片

输入:root = [1,1,1,null,1,null,null,1,1,null,1]

输出:4

解释:蓝色节点为树中最长交错路径(左 -> 右 -> 左 -> 右)。

示例 3:

输入:root = [1]

输出:0

提示:

  • 每棵树最多有 50000 个节点。

  • 每个节点的值在 [1, 100] 之间。

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

思路

采用深度优先遍历的方式,从顶向下访问到所有节点。并且遍历下一个子节点时,能够知道子节点是属于父节点的左子树,还是右子树。
每个节点定义两个变量, leftNum 与 rightNum ,分别代表当前节点为父节点的左子节点时的交错路径长度与当前节点为父节点的右子节点时的交错路径。
当然,当前节点只有两种可能:1 为左子节点 2 为右子节点

c++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:

    int result = 0;
    int longestZigZag(TreeNode* root) {
        if(root == nullptr) {
            return result;
        }

        dfs(root, 0, 0);

        return result;
    }

    void dfs(TreeNode* root, int leftNum, int rightNum){
        result = max(result, max(leftNum, rightNum));

        if(root->left != nullptr) {
            dfs(root->left, rightNum + 1, 0);
        }

        if(root->right != nullptr) {
            dfs(root->right, 0, leftNum + 1);
        }

    }


};

你可能感兴趣的:(二叉树,LeetCode,leetcode,算法,java)