如何在树未完成的情况下找到给定深度的节点值的总和?

时间:2010-07-13 19:59:23

标签: java binary-tree depth

之前我问过非常类似的问题,应该提到更详细的问题。 上一次是“如何在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;
        }

    }

2 个答案:

答案 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”,让它返回总和,不管前面条件是否成功,它应该有效。