单值二叉树

单值二叉树

class Solution {
    List vals;
    public boolean isUnivalTree(TreeNode root) {
        vals = new ArrayList();
        dfs(root);

        //如果vals list中有一个值跟第一个值不同，那就说明该树不是单值二叉树，返回false
        for (int val : vals) {
            if (val != vals.get(0))
                return false;
        }

        return true;
    }

    //将二叉树的每个结点的值加入到list中
    public void dfs(TreeNode node) {
        if (node != null) {
            vals.add(node.val);
            dfs(node.left);
            dfs(node.right);
        }
    }
}