
这一题,其实还是很简单的
就是一个自下往上的递归的过程。之所以是自下往上,是因为只要一颗子树不是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;
}
}
本文介绍了一种自底向上的递归算法,用于计算一棵树中所有节点值相同的子树的数量。通过判断每个节点的左右子树是否满足统一值条件,并递归地将结果向上反馈。
1820

被折叠的 条评论
为什么被折叠?



