【MAC 上学习 C++】Day 55-7. 实验11-2-8 单链表结点删除 (20 分)

实验11-2-8 单链表结点删除 (20 分)

1. 题目摘自

https://pintia.cn/problem-sets/13/problems/609

2. 题目内容

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

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

函数接口定义：

struct ListNode *readlist();
struct ListNode *deletem( struct ListNode *L, int m );
函数readlist从标准输入读入一系列正整数，按照读入顺序建立单链表。当读到−1时表示输入结束，函数应返回指向单链表头结点的指针。

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

输入样例：

10 11 10 12 10 -1
10

输出样例：

11 12

3. 源码参考

#include 
#include 

using namespace std;

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();
    cin >> m;
    L = deletem(L, m);
    printlist(L);

    return 0;
}

struct ListNode *readlist()
{
  struct ListNode *p, *h, *t;
  int n;

  h = NULL;
  cin >> n;
  while(n != -1)
  {
    p = (struct ListNode*)malloc(sizeof(struct ListNode));
    
    p->data = n;
    p->next = NULL;
    
    if(h == NULL)
    {
      h = p;
    }
    else
    {
      t->next = p;
    }
    
    t = p;
    cin >> n;
  }
  
  return h;
}

struct ListNode *deletem( struct ListNode *L, int m )
{
  struct ListNode *p;

  while((L != NULL) && (L->data == m))
  {
    L = L->next;
  }
  
  p = L;
  while(p->next)
  {
    if(p->next->data  == m)
    {
      p->next = p->next->next;
    }
    else
    {
      p = p->next;
    }
  }

  return L;
}