您的位置:首页 > 产品设计 > UI/UE

250. Count Univalue Subtrees

2016-10-14 07:21 246 查看
这个题做得不太好,上来思路就有了,但是逻辑上老犯错,我真是猪脑。

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.

需要传上的信息是:

1) 我是不是unique-value-subtree.

2) 没了。或者说我的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;

}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: