从单链表中删除元素

时间:2015-01-19 13:06:12

标签: c algorithm list pointers

我想从函数中的value指定的列表中删除一些元素。如果函数的功能不正常,我的功能就不起作用了。等于列表中的第一个元素。否则它运作良好。有什么想法吗?

struct elem {
    int val;
    struct elem *next;
};

void del(struct elem *list, int val) {
    struct elem* tmp = list;
    struct elem* prev = NULL;

    while (tmp != NULL) {
        if (tmp->val == val) {
            if (prev == NULL) {
                tmp = tmp->next;
                free(list);
                list = tmp;
            } else {
                prev->next = tmp->next;
                free(tmp);
                tmp = prev->next;
            }
        } else {
            prev = tmp;
            tmp = tmp->next;
        }
    }
}

1 个答案:

答案 0 :(得分:5)

您的通话功能无法知道list已更新。它甚至会继续引用已被删除的list。这不好。

一种解决方案是将列表作为struct elem **list传递:

void del(struct elem **list, int val) {
    struct elem* tmp = *list;
    struct elem* prev = NULL;

    while (tmp != NULL) {
        if (tmp->val == val) {
            if (prev == NULL) {
                tmp = tmp->next;
                free(*list);
                *list = tmp;
            } else {
                prev->next = tmp->next;
                free(tmp);
                tmp = prev->next;
            }
        } else {
            prev = tmp;
            tmp = tmp->next;
        }
    }
}

修改:还有其他解决方案。您可以返回新的列表指针:

struct elem *del(struct elem *list, int val) { ... }

你这样称呼它:

list = del(list, 12);

此解决方案的缺点是list在调用中有点多余,省略返回值是合法的,因此实际上不会更新列表。

我喜欢的解决方案是为列表定义控件结构。目前,它只包含头指针:

struct list {
    struct elem *head;
};

您在列表上运行的函数然后将指向此结构的指针作为参数:

void del(struct list *list, int val) {
    struct elem* tmp = list->head;
    struct elem* prev = NULL;

    while (tmp != NULL) {
        if (tmp->val == val) {
            if (prev == NULL) {
                tmp = tmp->next;
                free(list->head);
                list->head = tmp;
            } else {
                prev->next = tmp->next;
                free(tmp);
                tmp = prev->next;
            }
        } else {
            prev = tmp;
            tmp = tmp->next;
        }
    }
}

struct list可以有其他字段,例如尾部指针可以快速追加到末尾。您还可以跟踪列表长度。