【leetcode】5338. 二叉树中的最长交错路径( Longest ZigZag Path in a Binary Tree)

最新推荐文章于 2023-07-30 09:28:08 发布

Liber-coder

最新推荐文章于 2023-07-30 09:28:08 发布

阅读量318

点赞数

分类专栏： Leetcode 文章标签： leetcode 二叉树

本文链接：https://blog.csdn.net/a435262767/article/details/104735475

版权

Leetcode 专栏收录该内容

82 篇文章 6 订阅

订阅专栏

题目描述

【leetcode】5338. 二叉树中的最长交错路径( Longest ZigZag Path in a Binary Tree)

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

第一次解答

思路
深度优先遍历，注意不一定要从根结点开始。
一开始想：要判断最大，所以路径长度不是在递归最深处给出，而是从最深处不断返回到根节点，才能判断是否最大。
后来转念一想：因为路径起始点不一定在根节点，所以还得在递归最深处取数据。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
#include <cmath>
class Solution {
public:
   void helper(TreeNode* root, bool is_left, int &max_path, int curr_path){
        if(nullptr == root){
            return;
        }
        
        curr_path += 1;
        if(curr_path > max_path){
            max_path = curr_path;
        }
        
        TreeNode *p_next = (is_left ? root->left : root->right);
        helper(p_next, !is_left, max_path, curr_path); // 按规则，继续走折线
        
        TreeNode *p_next2 = (is_left ? root->right : root->left);
        helper(p_next2, is_left, max_path, 0); // 不按规则，重新走折线
        
        return;
    }
    int longestZigZag(TreeNode* root) {
        if(nullptr == root)
            return 0;
        int max_path = 0;
        helper(root->right, true, max_path, 0);   // L
        helper(root->left, false, max_path, 0);  // R
        return max_path;
    }
};