单值二叉树

/*
 * 思路：
 * 1. 使用Java Stream API来遍历二叉树。
 * 2. 将树的所有节点值收集到一个列表中。
 * 3. 检查列表中的所有值是否相同。
 */

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

public class Solution {
    public boolean isUnivalTree(TreeNode root) {
        if (root == null) return true;
        List<Integer> values = new ArrayList<>();
        collectValues(root, values);
        return values.stream().distinct().count() == 1;
    }

    private void collectValues(TreeNode node, List<Integer> values) {
        if (node == null) return;
        values.add(node.val);
        collectValues(node.left, values);
        collectValues(node.right, values);
    }
}