二叉树级别总和-

Question

一个基本的Java示例：给定一个二叉树和一个整数，它是目标级别的深度。 并计算目标级别中节点的总和。

我将函数中的值添加到左侧和右侧节点。 为什么这种解决方案在这种情况下不起作用，有人可以帮忙解释一下吗？

同样，当travese函数返回时，它返回父根还是更像break; 在循环中，流程停止了吗？

private int sum;

public int levelSum(TreeNode root, int level) {
    sum = 0;
    traverse(root, 1, level, 0);
    return sum;
}

public void traverse(TreeNode root, int depth, int level, int sum) {
    if(root == null) {
        return;
    }

    if(depth == level) {
        return;
    }

    if(root.left != null) {
        traverse(root.left, depth + 1, level, sum + root.left.val);
    }

    if(root.right != null) {
        traverse(root.right, depth + 1, level, sum + root.right.val);
    }
}

Answer 1

首先，您没有返回求和的结果。 您将不得不退还它。

然后，由于您写的是“在目标级别”，因此我想您只需要总结目标级别中的节点，而不是该级别之前的节点。

尝试这个：

public int levelSum(TreeNode root, int level) {
    return traverse(root, 1, level);

}

public void traverse(TreeNode root, int depth, int level){
    int sum = 0;

    if(root == null || depth > level){
        return 0;
    }

    if (depth == level) sum += root.val;

    if(root.left != null) {
        sum += traverse(root.left, depth + 1, level);
    }

    if(root.right != null) {
        sum += traverse(root.right, depth + 1, level);
    }

    return sum;
}