如果二叉树每个节点都具有相同的值,那么该二叉树就是单值二叉树。
只有给定的树是单值二叉树时,才返回 true
;否则返回 false
。
示例 1:
输入:[1,1,1,1,1,null,1]
输出:true
示例 2:
输入:[2,2,2,5,2]
输出:false
提示:
- 给定树的节点数范围是
[1, 100]
。 - 每个节点的值都是整数,范围为
[0, 99]
。
思路
主要的思路就是遍历一遍整棵二叉树,然后每次访问当前结点时,都记录在HashMap
中,其中,key
是当前结点的val
值,value
是该值在整棵树中出现的次数。
当遍历完整棵树之后,再查看map
中keySet
集合的大小,如果大于等于2
,则说明不是单值树,返回false
,否则返回true
即可。
代码
public class Solution {
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
public void dfs(TreeNode root) {
if(map.get(root.val)==null) {
map.put(root.val, 1);
}
else {
int value = map.get(root.val);
value++;
map.put(root.val, value);
}
if(root.left!=null) {
dfs(root.left);
}
if(root.right!=null) {
dfs(root.right);
}
}
public boolean isUnivalTree(TreeNode root) {
dfs(root);
if(map.keySet().size()>=2) {//如果有多个key
return false;
}
else return true;
}
}