Question

我很困惑为什么我在BST中移除叶子的方法不起作用。如果我将0分配给数据，它会反映在树中，但是当我为节点指定null时，它仍然能够在BST遍历中被引用。

public void removeLeaves(){
    removeLeaves(getRoot());
}

private void removeLeaves(IntTreeNode node) {
    if (node == null)
        return;
    else if( node.left == null && node.right == null){
        node.data = 0;  //im finding the leave nodes correctly
        node = null;    //but its not removing them
    }
    else {
        removeLeaves(node.left);
        removeLeaves(node.right);
    }
}

        overallRoot
        ____[7]____
       /           \
    [3]             [9]
   /   \           /   \
[0]     [0]     [0]     [8]
                           \
                            [0]

有人可以解释为什么这不按预期工作吗？

Answer 1

在示例树中，请考虑9

 9.left => null
 9.right => address of 8

当您分配node.data = 0;时，节点的地址为8，因此0将反映在树中。

但是当你node =null时，你只是在改变变量node。你没有对address of 8进行任何操作。

我认为你希望发生的事情是node = null：

      address of 8 = null

这实际上是不可能的，因为您只是更改变量node。

说8的地址是0XABCD，所以node = 0XABCD。如果node.data=0 node的地址为0XABCD，则0XABCD.data将更改为0。但是当您执行node = null时，您只是为变量node分配了一个新值，而您没有对原始地址0XABCD执行任何操作。

你实际需要做的是

  if(node.left!= null && node.left.left == null && node.left.right ==null)
     node.left =null

  if(node.right!= null && node.right.left == null && node.right.right ==null)
     node.right =null

<强>更新

你要做的是这样的事情：

  Foo foo = new Foo();
  Foo anotherFoo = foo;

  anotherFoo.value = something; // both foo.value and anotherFoo.value will be changed to "something", because of the reference.

  anotherFoo = null; 
  // here you are expecting foo also to be null which is not correct.

Answer 2

public void removeLeaves () {
    if (getRoot() != null) 
        removeLeaves (getRoot());
}

private IntTreeNode removeLeaves (IntTreeNode node) {
    if (getRoot().left == null && getRoot().right == null) {
        setRoot(null);
    } else {
        if (node.left != null) {
            if (node.left.left == null && node.left.right == null) 
                node.left = null;             
            else 
                removeLeaves(node.left);
        }
        if (node.right != null) {
            if (node.right.right == null && node.right.left == null) 
                node.right = null;      
            else 
                removeLeaves(node.right);     
        }
    }
    return node;
}

去除二叉树中的叶子

2 个答案: