0

我想要一些关于 Go 指针接收器如何工作的帮助。

我在下面有一个包含二叉搜索树的示例，希望能帮助我解释。

package main

import "fmt"

type Node struct {
  key         int
  left, right *Node
}

func NewNode(key int) *Node {
  return &Node{key, nil, nil}
}

type BST struct {
  root *Node
}

func NewBinarySearchTree() *BST {
  return &BST{nil}
}

func (t *BST) Insert(key int) {
  if t.root == nil {
    t.root = NewNode(key)
    return
  }
  var node = t.root
  for {
    if key < node.key {
      if node.left == nil {
        node.left = NewNode(key)
        return
      } else {
        node = node.left
      }
    } else {
      if node.right == nil {
        node.right = NewNode(key)
        return
      } else {
        node = node.right
      }
    }
  }
}

func inorder(node *Node) {
  if node == nil {
    return
  }
  inorder(node.left)
  fmt.Print(node.key, " ")
  inorder(node.right)
}

func main() {
  tree := NewBinarySearchTree()
  tree.Insert(3)
  tree.Insert(1)
  tree.Insert(2)
  tree.Insert(4)
  inorder(tree.root) // 1 2 3 4
}

然而，在我写完这个之后，我想我可以将我的插入函数简化如下：

func (t *BST) Insert2(key int) {
  var node *Node
  node = t.root
  for node != nil {
    if key < node.key {
      node = node.left
    } else {
      node = node.right
    }
  }
  node = NewNode(key)
}

但是，这样做树永远不会更新。我的想法是...

  • 在第一次插入时，根节点将为 nil。
  • 所以引用 t.root 的局部变量节点也将为 nil
  • 因此将跳过 for 循环。
  • node = NewNode(key)将具有相同的效果t.root = NewNode(key)

我的 Insert2 方法哪里出错了？有没有办法可以调整？

4

2 回答 2

3

您似乎对指针的使用感到困惑。

当你这样做时node = t.root，你只是node指向任何t.root指向的东西。

稍后，当您这样做时node = NewNode(key)，您将node指向一个新创建的项目，这不是您想要的；你想t.root指出那个新项目。

由于您打算修改类型为*Node( root,left和right) 的变量，我们需要一个指向它们的指针，因此需要一个类型为 的变量**Node，并具有更多的间接级别。

您可以首先node指向 , 的地址t.root，node := &t.root然后继续进行循环。

您可以尝试以下方法：

func (t *BST) Insert3(key int) {
    node := &t.root
    for *node != nil {
        if key < (*node).key {
            node = &(*node).left
        } else {
            node = &(*node).right
        }
    }
    *node = NewNode(key)
}

请注意，我们使用间接运算符*访问引用的数据，当检查循环上的地址时，还有键。

在函数的最后，*node = NewNode(key)做你原本打算做的事情；您正在将新创建的项目分配给根、左或右指针。

于 2013-11-07T09:00:16.097 回答
3 
node = NewNode(key)

那条线不会改变树。那行改变了局部变量node；在这条线之后，node指向不同的节点，但它曾经指向的对象不受影响。要插入树中，您必须分配给t.root、node.left或node.right。

于 2013-11-06T21:15:35.257 回答