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;
        
    }
}

转载于:https://www.cnblogs.com/reboot329/p/5958946.html