单链表结点删除

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

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 

思路：注意首结点data等于m的情况，此时pre结点还未接入链表，故要特殊处理。

struct ListNode *readlist() {
    struct ListNode *head = NULL, *p, *pre;//pre保存当前结点的前驱结点
    int n;
    scanf("%d", &n);
    while (n != -1) {
        p = (struct ListNode *) malloc(sizeof(struct ListNode));
        p->data = n;
        p->next = NULL;//新结点的指针域设为NULL
        if (head == NULL)
            head = p;
        else
            pre->next = p;//前驱结点的指针域设为当前新建结点的地址
        pre = p;//把pre设为p，作为下个结点的前驱结点
        scanf("%d", &n);
    }
    return head;//返回头结点
}

struct ListNode *deletem(struct ListNode *L, int m) {
    struct ListNode *p, *pre;//pre保存当前结点的前驱结点
    p = L, pre = NULL;
    while (p) {
        if (p->data == m) {
            if (pre) {//情况1：pre！=NULL时，说明首结点的data不等于m
                pre->next = p->next;//结点指针后移
                free(p);
                p = pre->next;
            } else {//情况2：pre==NULL时，说明首结点的data等于m
                L = p->next;//链表首结点指针后移
                free(p);
                p = L;
            };
        } else {//data不等于m，pre和p均向后移一位
            pre = p;
            p = p->next;
        }
    }
    return L;
}