LeetCode题解之Univalued Binary Tree

1、题目描述

2、问题分析

遍历一遍树，然后将所有节点的数值放入到一个set中，最后检查set中元素的个数是否为1.

3、代码

bool isUnivalTree(TreeNode* root) {
        set<int> s;
        preOrder(root, s);
        return s.size() == 1;
    }
    
    void preOrder(TreeNode* root, set<int> &s)
    {
        if (root == NULL)
            return;
        s.insert(root->val);
        preOrder(root->left, s);
        preOrder(root->right,s);
    }