zoukankan      html  css  js  c++  java
  • 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;
            
        }
    }
    
  • 相关阅读:
    Debian vim没有颜色的解决办法
    第四次作业
    第三次作业
    第二次作业
    Mad Libs游戏 & 华氏温度与摄氏温度转换
    有趣故事
    Mybatis的使用及增删改查
    jdbc数据连接池
    详解MySQL数据类型
    SQL语法大全
  • 原文地址:https://www.cnblogs.com/reboot329/p/5958946.html
Copyright © 2011-2022 走看看