从原始树中删除深度k以下的所有子树？

Question

我完全被这个问题困住了，不知道从哪里开始。我给它的每一个测试它都失败了，但我不知道如何修复它。指令是： 从原始树中删除深度k以下的所有子树

相关信息： - 不要更改节点类。 - 不要更改任何函数的第一行：名称、参数、类型。 您可以添加新功能，但不要删除任何内容
- 函数必须是递归的 - 没有循环 - 每个函数必须只有一个递归辅助函数，您可以添加它 - 每个函数必须是独立的 --- 不要调用除了助手之外的任何函数 - 没有字段（在函数之外声明的变量）

如果您认为我遗漏了任何信息/代码，请告诉我。

相关代码：

private Node root;
private static class Node {
    public final int key;
    public Node left, right;
    public Node(int key) { this.key = key; }
}



public void removeBelowDepth(int k) {
    removeBelowDepthHelper(root, 0, k);
    }
private void removeBelowDepthHelper(Node node, int currentDepth, int k) {
    if (node == null) return;
    if (currentDepth == k) {
        node = null;
        return;
    }
    removeBelowDepthHelper(node.left, currentDepth + 1, k);
    removeBelowDepthHelper(node.right, currentDepth + 1, k);
    }

Answer 1

我接受了 AJ 的建议，只是更改了 null 调用，它完美地工作！非常感谢你们！

public void removeBelowDepth(int k) {
        removeBelowDepthHelper(root, 0, k);
        }
        private void removeBelowDepthHelper(Node node, int currentDepth, int k) {
        if (node == null) return;
        if (currentDepth == k) {
            node.left = null;
            node.right = null;
            return;
        }
        removeBelowDepthHelper(node.left, currentDepth + 1, k);
        removeBelowDepthHelper(node.right, currentDepth + 1, k);
        }