[Leetcode] 250. Count Univalue Subtrees

这一题,其实还是很简单的
就是一个自下往上的递归的过程。之所以是自下往上,是因为只要一颗子树不是univalue的,那么任何包含这颗子树的树都不是univalue的。同样的,uni value本身也是具备传递性的,也就是说如果某一个子树是univalue的,那么它的父亲节点只需要和这个子树的根节点进行比较就可以了,因为整个树的值都是一样的。

所以整个算法基本就是,先往左右要结果,满足以下条件的为univalue subtree
1. 左子树为空或者左子树为univalue subtree并且当前节点和左子树根节点值相同
2. 右子树为空或者右子树为univalue subtree并且当前节点和左子树根节点值相同
空节点我们也可以认为是一个univalue subtree。
当一个子树被判定为univalue subtree的时候就在计数器加一即可。

给出代码如下:

    public int countUnivalSubtrees(TreeNode root) {
        int[] result = new int[1];
        countHelper(root, result);
        return result[0];
    }

    public boolean countHelper(TreeNode root, int[] result) {
        if (root == null) {
            return true;
        } else {
            boolean left = countHelper(root.left, result);
            boolean right = countHelper(root.right, result);
            boolean isUni = (left && (root.left == null || root.left.val == root.val))
                            && (right && (root.right == null || root.right.val == root.val));
            if (isUni) result[0]++;
            return isUni;
        }
    }

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值