给你一个二叉树的根节点 root ,计算并返回 整个树 的坡度 。
一个树的 节点的坡度 定义即为,该节点左子树的节点之和和右子树节点之和的 差的绝对值 。如果没有左子树的话,左子树的节点之和为 0 ;没有右子树的话也是一样。空结点的坡度是 0 。
整个树 的坡度就是其所有节点的坡度之和。
一步到位
class Solution {
int ans;
public int findTilt(TreeNode root) {
dfs(root);
return ans;
}
int dfs(TreeNode root) {
if (root == null) return 0;
int l=dfs(root.left),r=dfs(root.right);
ans+= Math.abs(l-r);
return l+r+root.val;
}
}
求和与坡度计算分开
public int findTilt(TreeNode root) {
if (root == null) return 0;
return findTilt(root.left)+findTilt(root.right)+Math.abs(getSum(root.left)-getSum(root.right));
}
int getSum(TreeNode root) {
if (root == null) return 0;
return getSum(root.left) + getSum(root.right) + root.val;
}