这个题做得不太好,上来思路就有了,但是逻辑上老犯错,我真是猪脑。
traverse的时候,判断一个node是不是uni-value tree的时候有2点:
他的左右子树是uni-value tree;
他和左右子树的值相等。
缺一不可。。
一旦是uni-value tree,总数++就行了。。
但是比较子树VALUE的时候牵扯到子树是NULL的情况,还要多加个判断不是NULL的情况。一开始犯贱想写成一行,逻辑乱了,装逼失败,惨遭打脸。
public class Solution
{
int res = 0;
public int countUnivalSubtrees(TreeNode root)
{
if(root == null) return 0;
if(root.left == null && root.right == null) return 1;
helper(root);
return res;
}
public boolean helper(TreeNode root)
{
if(root == null) return true;
boolean left = helper(root.left);
boolean right = helper(root.right);
if(left && right)
if((root.left == null || root.left.val == root.val) && (root.right ==null || root.right.val == root.val))
res++;
else return false;
return left && right;
}
}
/*
[1,-2,3,-2,null,null,null,null,5]
[5,1,5,5,5,null,5]
[5,1,5,5,5,1,1]
[5,5,5,5,5,5,5]
[7,82,82,-79,98,98,-79,-79,null,-28,-24,-28,-24,null,-79,null,97,65,-4,null,3,-4,65,3,null,97]
*/
二刷。
算多少个所有val都相等的子树。 比如[2,2]本身是1个,然后leaf又是一个,所以是2个,给的例子很好得阐述了是怎么回事。。
还是和上一个一样,bottom-up via post order traversal.
需要传上的信息是:
- 我是不是unique-value-subtree.
- 没了。或者说我的root.val,但是可以直接通过上面的root.left.val或者root.right.val来获取,所以不算是传上去的。
所以只有1)这一个信息,那就不用建新class了。
和今天做的2个有点像,124和333,但是比那俩简单。
一开始做错了,以为是算总node的和,其实没那么复杂,就算总subTree的和就行。
Time: O(n)
Space: O(n)
public class Solution {
int res = 0;
public int countUnivalSubtrees(TreeNode root) {
if (root == null) return 0;
postOrder(root);
return res;
}
public boolean postOrder(TreeNode root) {
if (root == null) return true;
boolean left = postOrder(root.left);
boolean right = postOrder(root.right);
if (left && right) {
if ((root.left == null || root.val == root.left.val) &&
(root.right == null || root.val == root.right.val)) {
res ++;
return true;
}
}
return false;
}
}