二叉树的直径

/*
 * 题目思路：
 * 使用Java Stream流来计算二叉树的直径。
 * 通过流操作递归计算左右子树的深度，更新最大直径。
 */
 
import java.util.stream.Stream;
 
public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}
 
public class Solution {
    private int maxDiameter = 0;
 
    public int diameterOfBinaryTree(TreeNode root) {
        calculateDepth(root);
        return maxDiameter;
    }
 
    private int calculateDepth(TreeNode node) {
        if (node == null) return 0;
        int leftDepth = Stream.of(node.left).mapToInt(this::calculateDepth).sum();
        int rightDepth = Stream.of(node.right).mapToInt(this::calculateDepth).sum();
        // 更新最大直径
        maxDiameter = Math.max(maxDiameter, leftDepth + rightDepth);
        // 返回当前节点的深度
        return Math.max(leftDepth, rightDepth) + 1;
    }
}