之前我问过非常类似的问题,应该提到更详细的问题。 上一次是“如何在PHP中找到二叉树中给定深度的节点值的总和”。
sum(Node, Level) =
if (Level == 0) return Node.value;
else return f(Node.left, Level-1) +
f(Node.right, Level-1).
所以现在我试着用Java写这个。并且Java抛出nullPointerException。原因是因为如果树不完整,下面的代码不会处理。
public int getNodeValueByDepth(Node n, int level) {
if(level == 0) {
return n.data;
}
else {
return getNodeValueByDepth(n.left, level-1) +
getNodeValueByDepth(n.right, level-1);
}
}
我的测试树结构是:
/*
* construct tree
* sum of node's value
* 5 depth 0 ==> 5
* / \
* 3 10 depth 1 ==> 13
* / \ / \
* 2 4 6 11 depth 2 ==> 23
* / \
* 7 9 depth 3 ==> 16
*
* depth 4 ==> null
*
*/
因此,当我调用getNodeValueByDepth(root,3)为7 + 9时,它会抛出null指针异常错误。我试图添加逻辑来处理节点左和右为空的情况,但仍然无法弄清楚如何解决这个问题我无法入睡。
有人能给我一个提示吗?我试过但不是它只返回0.
public int getNodeValueByDepth(Node n, int level) {
int sum = 0;
if(level == 0) {
return sum + n.data;
}
else if(n.left != null) {
return sum += getNodeValueByDepth(n.left, level-1);
}
else if(n.right != null) {
return sum += getNodeValueByDepth(n.right, level-1);
}
else {
return sum + 0;
}
}
答案 0 :(得分:1)
您的控制语句在每种情况下都使用else
,因此只有一个会被评估。因此,如果它评估左侧,当它返回时它不会做右侧。
public int getNodeValueByDepth(Node n, int level) {
int sum = 0;
/** If we have reached our desired level */
if (level == 0) {
if (n != null) {
/** Assuming data is an int and not nullable */
return n.data;
} else {
/** Change this to 0 if you don't want an error condition */
return -1;
}
/** We are not the desired level, so get the sum of .left and .right, knowing either may not exist */
if (n.left != null) {
sum += getNodeValueByDepth(n.left, level-1);
}
if (n.right != null) {
sum += getNodeValueByDepth(n.right, level-1);
}
/** Now have evaluated every possible child and stored the sum, even if it is 0 */
return sum;
}
注意:检查语法错误,我在没有Java的机器上写了这个。
答案 1 :(得分:1)
在最后一个例子中,取出最后一个“Else”,让它返回总和,不管前面条件是否成功,它应该有效。