链接:
https://leetcode.com/problems/binary-tree-tilt/
大意:
对于一棵树,定义一个概念-----树节点的tilt:为该节点下左子树所有节点的值与右子树所有节点的值之间的绝对值。给定树的根节点,求该树所有节点的tilt的和。例子:(空节点的tilt为0)
思路:
递归。主要问题是怎么设计这个递归?首先,对于每个树节点,它需要保存两个变量:一个是该节点的tilt值(左子树所有节点之和和右子树所有节点之和的绝对值),另一个变量是以该节点为根节点所形成的树中所有节点的和(用于返回给其父节点,以便父节点计算自己的tilt)。在我的代码里,我是把所有节点的tilt值的和定义为了一个外部实例变量sum,而以每个节点为根节点所对应子树的所有节点的和可以由递归求得。具体方法如下:
getSum(root)返回以root为根节点时整棵树的所有节点值的和,那么它应该是等于 root.val + getSum(root.left) + getSum(root.right) 。而此时需要考虑一些特殊情况,如:root.left和root.right都为空时,那么应该返回的就是root.val;如果root.left 和 root.right只有一个为空时(假设root.left为null),则返回的应是root.val + getSum(root.right)。而在返回这个结果之前,我们应该先计算一下该节点的tilt值。该节点的tilt值计算为 Math.abs(getSum(root.left) - getSum(root.right)) 。计算总的tilt值只需用外部实例变量 sum 加上在本次递归计算得到的tilt值即可
代码:
class Solution {
private int sum = 0;
public int findTilt(TreeNode root) {
getSum(root);
return sum;
}
// 返回值为 以root为根节点的子树的所有节点值的和
public int getSum(TreeNode root) {
if (root == null)
return 0;
if (root.left == null && root.right == null)
return root.val;
if (root.left == null) {
int rSum = getSum(root.right);
sum += Math.abs(rSum);
return root.val + rSum;
}
if (root.right == null) {
int lSum = getSum(root.left);
sum += Math.abs(lSum);
return root.val + lSum;
}
int lSum = getSum(root.left), rSum = getSum(root.right);
sum += Math.abs(lSum - rSum);
return root.val + lSum + rSum;
}
}
结果:
结论:
使用递归时,没必要深入递归函数之内一步步执行,需要明白的是该递归函数完成的是什么功能即可!
另:感觉自己代码写的有点冗余......
最佳:
class Solution {
private int sum = 0;
public int findTilt(TreeNode root) {
getSum(root);
return sum;
}
// 返回值为 以root为根节点的子树的所有节点值的和
public int getSum(TreeNode root) {
if (root == null)
return 0;
if (root.left == null && root.right == null)
return root.val;
int lSum = getSum(root.left), rSum = getSum(root.right);
sum += Math.abs(lSum - rSum);
return root.val + lSum + rSum;
}
}
去掉的中间那一部分代码实际是可以由第一个跳出递归的条件来完成的。先思考再下手!