习题11-8 单链表结点删除

习题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的结点删除。返回指向结果链表头结点的指针。

问题：分别实现完成要求的两个程序接口后，在本地跑了下样例什么的都没问题，拿到PTA上就报错，不知道哪里出了问题，热心的朋友麻烦指点一下。

代码：

/*---------------------裁判程序------------------------*/
#include <stdio.h>
#include <stdlib.h>

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

struct ListNode *readlist(void);
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;
}
/*------------------需实现的程序接口-----------------*/
struct ListNode *readlist(void) {
    struct ListNode *head = NULL;
    struct ListNode *end = NULL;
    int number;
    scanf("%d",&number);
    do {
        struct ListNode *p = (struct ListNode*) malloc (sizeof(struct ListNode));
        if(p != NULL) {
            p->data = number;
            p->next = NULL;
        }
        else exit(1);
        if(head != NULL) {
            end->next = p;
        }
        else {
            head = p;
        }
        end = p;    //尾节点必为新加入的节点
        scanf("%d",&number);
    }while(number != -1);
    return head;
}

struct ListNode *deletem( struct ListNode *L, int m ) {
    struct ListNode *p = L;
    struct ListNode *q = NULL; //临时节点，用来保存删除节点指向的下一个节点
    while(p != NULL) {
        if(p->data == m) {     //找到该数
            if(q != NULL && p != L) {       //非头节点找到该数
                q->next = p->next;
                free(p);
            }
            else {                          //头节点处找到该数
                L = p->next;
            }
        }
        else {                 //未找到该数
            q = p;             //为了让q不为NULL
        }
        p = p->next;
    }
    
    return L;
}