Question

我试图编写一个迭代破坏二叉树的代码。我知道递归更容易，但我想我会尝试迭代地做（虽然主要使用C表示法并且只使用结构，不需要类）。我的想法如下：

void delete_tree (node *p_tree){
node *local_node;
node *parent;
int left_or_right;
while (p_tree->p_left != NULL || p_tree->p_right != NULL){
    parent = p_tree;
    if (parent->p_left != NULL){
        local_node = parent->p_left;
        left_or_right = 1;
    } else {
        local_node = parent->p_right;
        left_or_right = 2;
    }
    while (local_node->p_left != NULL || local_node->p_right != NULL){
        if (local_node->p_left == NULL){
            parent = local_node;
            local_node = local_node->p_right;
            left_or_right = 2;
        } else if (local_node ->p_right == NULL){
            parent = local_node;
            local_node = local_node->p_left;
            left_or_right = 1;
        } else {
            parent = local_node;
            local_node = local_node->p_left;
            left_or_right = 1;
        }
    }
    cout << "Deleting node with value: " << local_node->key_value << endl;
    delete local_node;
    local_node = NULL;
    if (left_or_right == 1){
        parent->p_left = NULL;
    } else if (left_or_right == 2){
        parent->p_right = NULL;
    }
}
cout << "Deleting node with value: " << p_tree->key_value << endl;
delete p_tree;
p_tree = NULL;

}

我不知道我的逻辑有什么问题。它是这样的。遍历二叉树，直到我们命中没有子节点的节点，然后删除该节点。重复，直到我们只有原始父节点。然后最后删除它。

编辑：我已根据建议更改了代码。它现在似乎工作并且没有无限循环，但是当我尝试打印出二叉树的内容时，该函数会陷入无限循环。我知道我的打印功能是正确的，所以这个delete_tree函数仍然有问题。

Answer 1

这里有一个错误：

while (local_node->p_left != NULL && local_node->p_left != NULL)

当节点有两个孩子时，你只是向下迭代。您想要检查||而不是&&。

Answer 2

我会用一堆指向节点的指针来做。有人喜欢这样：

void preOrderStack_aux(node *p)
{
  if (p == NULL) 
    return;

  stack_t stack;

  stack_push(p); 

  while (not stack_is_empty(stack))
    {
      p = stack_pop(stack);

      if (p->p_right != NULL)
        stack_push(p->p_right);

      if (LLINK(p) != NULL)
        stack_push(p->p_link);

      // here stack has stored descendents
      free(p);
    }
}

c ++迭代地销毁二叉树

2 个答案: