【头歌实训：移除链表元素】

头歌实训：移除链表元素

任务描述

本关任务：给你一个带附加头结点L的单链表和一个整数 val ，请你删除链表中所有值为 val 的结点。
例如，下图所示的单链表L，val=6，删除所有值为6的结点。

编程要求

根据提示，在右侧编辑器补充代码，完成函数void remove_elements(LinkNode *&L, ElemType val)，该函数功能为：删除L中所有值为val的结点。

测试说明

平台会对你编写的代码进行测试：

输入格式

输入包括两行。
第一行有两个空格分隔的整数。第一个整数为单链表结点数量n，第二个数为需要删除的元素val。
第二行为空格隔开的n个整数。

输出格式

输出包括一行。为删除val后的单链表，每个数据后一个空格。

样例输入

7 6
1 2 6 3 4 5 6

样例输出

L->1 2 3 4 5

开始你的任务吧，祝你成功！

源代码：

#include "linklist.h"

/**
 * 删除链表元素: 删除L中所有值为val的结点
 */
void remove_elements(LinkNode *&L, ElemType val)
{
	//请在下面编写代码
	/******************Begin******************/
	LinkNode *pre = L, *p = L->next; //建立指针pre指向头节点,p指向首元结点
    while(p)            //while循环遍历p
    {
        if(p->data != val)      //当值不为val时
        {
            pre->next = p;        //尾插法将值插入
            pre = p;              //未结点向后移动
        }
        p = p->next;            //p向后移动
    }
    pre->next = NULL;               //最后一个结点后面置为空
	/*******************End*******************/
}


//尾插法建立单链表
void CreateListR(LinkNode *&L, ElemType a[], int n)
{
	LinkNode *s, *r;
	L = (LinkNode *)malloc(sizeof(LinkNode));  	//创建头结点
	L->next = NULL;
	r = L;					//r始终指向终端结点,开始时指向头结点
	for (int i = 0; i < n; i++)
	{
		s = (LinkNode *)malloc(sizeof(LinkNode)); //创建新结点s
		s->data = a[i];
		r->next = s;		//将结点s插入结点r之后
		r = s;
	}
	r->next = NULL;			//终端结点next域置为NULL
}

//初始化单链表
void InitList(LinkNode *&L)
{
	L = (LinkNode *)malloc(sizeof(LinkNode));  	//创建头结点
	L->next = NULL;
}

//销毁单链表
void DestroyList(LinkNode *&L)
{
	LinkNode *pre = L, *p = pre->next;
	while (p != NULL)
	{
		free(pre);
		pre = p;
		p = pre->next;
	}
	free(pre);	//此时p为NULL,pre指向尾结点,释放它
}

//输出单链表
void DispList(LinkNode *L)
{
	if (L->next == NULL) {
		printf("NULL\n");
		return;
	}
	LinkNode *p = L->next;
	while (p != NULL)
	{
		printf("%d ", p->data);
		p = p->next;
	}
	printf("\n");
}