如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true
;否则返回 false
。
示例 1:
输入:[1,1,1,1,1,null,1]
输出:true
示例 2:
输入:[2,2,2,5,2]
输出:false
提示:
- 给定树的节点数范围是
[1, 100]
。 - 每个节点的值都是整数,范围为
[0, 99]
。
例如:
解法一:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isUnivalTree(TreeNode* root)
{
if(!root) return true;
int nums = root->val;
return dfs(root, nums);
}
bool dfs(TreeNode *root, int &nums)
{
if(!root) return true;
if(root->val != nums) return false;
return dfs(root->left, nums) && dfs(root->right, nums);
}
};
解法二:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
unordered_set<int> tmp;
bool isUnivalTree(TreeNode *root)
{
if (root)
{
tmp.insert(root->val);
isUnivalTree(root->left);
isUnivalTree(root->right);
}
return tmp.size() == 1;
}
};