[Leetcode] 250. Count Univalue Subtrees

最新推荐文章于 2022-03-03 18:57:01 发布

原创最新推荐文章于 2022-03-03 18:57:01 发布 · 416 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

294 篇文章

订阅专栏

本文介绍了一种自底向上的递归算法，用于计算一棵树中所有节点值相同的子树的数量。通过判断每个节点的左右子树是否满足统一值条件，并递归地将结果向上反馈。

这一题，其实还是很简单的
就是一个自下往上的递归的过程。之所以是自下往上，是因为只要一颗子树不是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;
        }
    }