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

猜你喜欢

转载自blog.csdn.net/chaochen1407/article/details/81272394