统计同值子树

/*
思路：
1. 使用Java Stream API来处理二叉树节点的后序遍历。
2. 为每个节点判断其左右子树是否为同值子树，并且是否与当前节点值相同。
3. 如果满足条件，则增加计数器。
4. 最终返回计数器值。
*/
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Predicate;
 
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}
 
public class Solution {
    public int countUnivalSubtrees(TreeNode root) {
        AtomicInteger count = new AtomicInteger(0);
        
        Predicate<TreeNode> isUnivalSubtree = new Predicate<TreeNode>() {
            @Override
            public boolean test(TreeNode node) {
                if (node == null) return true;
                boolean left = test(node.left);
                boolean right = test(node.right);
                if (left && right) {
                    if (node.left != null && node.left.val != node.val) return false;
                    if (node.right != null && node.right.val != node.val) return false;
                    count.incrementAndGet();
                    return true;
                }
                return false;
            }
        };
        
        isUnivalSubtree.test(root);
        return count.get();
    }
}