在递归 AVL 树插入期间维护父指针 C++

Question

我正在寻找一种方法来递归维护我的 AVL 树插入中的父指针。 我知道这不是通常的过程，但是我打算有一个 function 来手动旋转节点，所以父级是必要的。

我得到的最接近的是维护根的父节点和未指向 NULL 的节点。所有其他节点将只指向它们自己。 如果我取消注释我的其他父插入，我会得到段错误。

AVL::node *AVL::insert(int k, string d, node *&n)
{
    //recursive base  case, I imagine n->parent = n should not be here

    if (n == NULL) {
        n = new node;
        n->left = NULL;
        n->right = NULL;
        n->key = k;
        n->data = d;
        n->height = 1;
        n->parent = n;
        return n;
    }

    if (n->key > k) {
        if (n->left != NULL) {
            n->left->parent = n;
            insert(k, d, n->left);
        } else {
            //n->left->parent = n;
            // this creates SEGFAULT
            insert(k, d, n->left);
        }
    } else if (n->key < k) {
        if (n->right != NULL) {
            n->right->parent = n;
            insert(k, d, n->right);
    } else {
        //n->right->parent = n;
        // this creates SEGFAULT
        insert(k, d, n->right);
    }
  } else {
      return n;
  }

    // insert height management here

    updateHeight(n);
    return n;
}

Answer 1

你几乎明白了。 你错过的剩下的一块拼图是：

insert(k, d, n->left);

您传递对将成为指向 AVL 树中新创建节点的指针的引用的方法（在本例中为n->left ）是正确的方法。

您遗漏的部分是明显观察到新节点需要知道它的父节点，而您正盯着它，就在上面一行。 这是n ：

insert(k, d, n, n->left);

现在， insert()知道父节点（和另一个使用n->right的递归调用，做同样的事情）：

AVL::node *AVL::insert(int k, string d, node *parent, node *&n)

您可以完全清除所有试图摆弄parent的复杂逻辑，以及指针在递归调用之前/之后是否为nullptr 。 您的递归insert()以更简单的方式确切地知道父级是什么：

    if (n == NULL) {
        n = new node;
        n->left = NULL;
        n->right = NULL;
        n->parent = parent;

差不多就是这样，这变成了一个又大又肥又没什么的汉堡。 据推测，对insert()的初始调用，即开始使用 AVL 树的根节点将球滚下山，将为父参数传递一个nullptr ，一切都会自行解决。