二叉树最长连续序列 II

/*
 题目思路：
 1. 在Java Stream中处理二叉树的递归问题并不直观，因此主要的逻辑还是要依赖于递归函数。
 2. 我们可以使用Stream来简化部分代码的编写，但是整体结构与普通Java解决方案差异不大。
*/
 
public class Solution {
    private int maxLength = 0;
    
    public int longestConsecutive(TreeNode root) {
        longestPath(root);
        return maxLength;
    }
    
    private int[] longestPath(TreeNode root) {
        if (root == null) return new int[]{0, 0};
        
        int incr = 1, decr = 1;
        
        if (root.left != null) {
            int[] left = longestPath(root.left);
            if (root.val == root.left.val + 1) {
                decr = left[1] + 1;
            } else if (root.val == root.left.val - 1) {
                incr = left[0] + 1;
            }
        }
        
        if (root.right != null) {
            int[] right = longestPath(root.right);
            if (root.val == root.right.val + 1) {
                decr = Math.max(decr, right[1] + 1);
            } else if (root.val == root.right.val - 1) {
                incr = Math.max(incr, right[0] + 1);
            }
        }
        
        maxLength = Math.max(maxLength, incr + decr - 1);
        return new int[]{incr, decr};
    }
}