【Leetcode】1372. Longest ZigZag Path in a Binary Tree

题目地址：

https://leetcode.com/problems/longest-zigzag-path-in-a-binary-tree/

给定一棵二叉树，求其最长的形如"左右左右……”或者“右左右左……”下降路径长度。返回最长路径边数。

思路是DFS。每次遍历到一个节点的时候，就算一下从它向下的形如"左右左右……”和“右左右左……”的最长路径节点数都算出来，更新答案，并且返回上层。代码如下：

public class Solution {
    
    
    
    private int res;
    
    public int longestZigZag(TreeNode root) {
    
    
        res = 1;
        dfs(root);
        // res存的是最长路径节点数，减去1才是边数
        return res - 1;
    }
    
    // 返回从cur向左向右走的最大zigzag路径的节点数，
    // [0]代表向左走，[1]代表向右边走
    private int[] dfs(TreeNode cur) {
    
    
        if (cur == null) {
    
    
            return new int[]{
    
    0, 0};
        }
        
        int[] left = dfs(cur.left), right = dfs(cur.right);
        res = Math.max(res, 1 + left[1]);
        res = Math.max(res, 1 + right[0]);
        
        return new int[]{
    
    1 + left[1], 1 + right[0]};
    }
}

class TreeNode {
    
    
    int val;
    TreeNode left, right;
    
    public TreeNode(int val) {
    
    
        this.val = val;
    }
}

时间复杂度$O(n)$，空间$O(h)$。