Question

我编写了使用队列遍历树的代码，但下面的出列函数会产生错误，head = p->next是否有问题？我无法弄清楚为什么这部分是错误的。

void Levelorder(void) {
node *tmp, *p;


if (root == NULL) return;

tmp = root;
printf("The level order is :\n");

while (tmp != NULL) {

    printf("%d, ", tmp->data);
    if (tmp->left) {
        enqueue(tmp->left);
    }
    if (tmp->right) {
        enqueue(tmp->right);
    }
    tmp = dequeue();
}

return;
}

void enqueue(node *p) {
if (head == NULL) {
    head = p;
}
else {
    tail->next = p;
}
tail = p;
p->next = NULL;
tail->next = NULL;

return;
}

node* dequeue(void) {
node *p;
p = head;
head = p->next;


if (head == NULL) {
    tail == NULL;
}

return p;
}

Answer 1

你的while循环的条件是：

while (tmp != NULL) {

因此，仅当dequeue在此处返回NULL时才会终止：

    tmp = dequeue();

但是在查看dequeue的实现时，这不可能发生：

node* dequeue(void) {
    node *p;
    p = head;

此处，p已取消引用：

    head = p->next;

    if (head == NULL) {
        tail == NULL;
    }

在这里，返回p：

    return p;
}

要返回NULL指针并离开while循环，p必须在NULL。但是，NULL指针之前会被head = p->next;取消引用，这会导致分段错误（就C语言而言是UB）。

如果head是NULL指针，你应该在你的dequeue-function的开头检查，在这种情况下返回NULL：

node* dequeue(void) {
    node *p;
    if (!head)
         return NULL;

...

树：级别顺序遍历使用队列

1 个答案: