250. Count Univalue Subtrees

这个题做得不太好,上来思路就有了,但是逻辑上老犯错,我真是猪脑。

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;
        
    }
}
posted @ 2016-10-14 07:21  哇呀呀..生气啦~  阅读(127)  评论(0编辑  收藏  举报