最长同值路径

/*
 * Java Stream 不适用于处理递归和二叉树的问题，所以这里还是使用传统方法。
 * 使用递归方法遍历二叉树。
 * 对于每个节点，计算其左右子树中与当前节点值相同的最长路径。
 * 更新全局变量记录最长路径长度。
 */
 
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}
 
class Solution {
    private int maxLength = 0;
 
    public int longestUnivaluePath(TreeNode root) {
        if (root == null) return 0;
        dfs(root);
        return maxLength;
    }
 
    private int dfs(TreeNode node) {
        if (node == null) return 0;
        int left = dfs(node.left);
        int right = dfs(node.right);
        int leftPath = 0, rightPath = 0;
        if (node.left != null && node.left.val == node.val) {
            leftPath = left + 1;
        }
        if (node.right != null && node.right.val == node.val) {
            rightPath = right + 1;
        }
        maxLength = Math.max(maxLength, leftPath + rightPath);
        return Math.max(leftPath, rightPath);
    }
}