习题11-8 单链表结点删除 (20分)

习题11-8 单链表结点删除 (20分)

本题要求实现两个函数，分别将读入的数据存储为单链表、将链表中所有存储了某给定值的结点删除。链表结点定义如下：

struct ListNode {
    int data;
    ListNode *next;
};

函数接口定义：

struct ListNode *readlist();
struct ListNode *deletem( struct ListNode *L, int m );

函数readlist从标准输入读入一系列正整数，按照读入顺序建立单链表。当读到−1时表示输入结束，函数应返回指向单链表头结点的指针。

函数deletem将单链表L中所有存储了m的结点删除。返回指向结果链表头结点的指针。

裁判测试程序样例：

#include <stdio.h>
#include <stdlib.h>

struct ListNode {
    int data;
    struct ListNode *next;
};

struct ListNode *readlist();
struct ListNode *deletem( struct ListNode *L, int m );
void printlist( struct ListNode *L )
{
     struct ListNode *p = L;
     while (p) {
           printf("%d ", p->data);
           p = p->next;
     }
     printf("\n");
}

int main()
{
    int m;
    struct ListNode *L = readlist();
    scanf("%d", &m);
    L = deletem(L, m);
    printlist(L);

    return 0;
}

/* 你的代码将被嵌在这里 */

输入样例：

10 11 10 12 10 -1
10

输出样例：

11 12 

我一开始的时候在 连续从头、中、尾删除 这个测试用例过不去，后来想想指向现在要删除的这个结点的前一个结点如果是要删除的呢？或者已经 free() 了呢？

如果你也是第四个过不去试试：

1 2 1 1 1 2 1 -1
1

struct ListNode *readlist()
{
    struct ListNode *head, *tail, *p;
    int n;
    head = tail = p = NULL;
    do {
        scanf("%d", &n);
        if (n < 0) {
            break;
        } else {
            p = malloc(sizeof(struct ListNode));
            p->data = n;
            p->next = NULL;
            if (tail != NULL) {
                tail->next = p;
            } else {
                head = p;
            }
            tail = p;
        }
    } while (n > 0);
    return head;
}

struct ListNode *deletem( struct ListNode *L, int m )
{
    struct ListNode *p, *q;
    for (p = L, q = NULL; p != NULL; p = p->next) {
        if (p->data == m) {
            if ( q && p != L) {
                q->next = p->next;
            } else {
                L = p->next;
            }
            free(p);
        } else {
            q = p;
        }
    }
    return L;
}