题目:给你一个链表的头节点 head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点 。
示例:
输入: | head = [1,2,6,3,4,5,6], val = 6 |
输出 | [1,2,3,4,5] |
解题思路:
- head 为空, 则直接返回head;
- head 内所有数据均等于val,则返回空;
- 如果链表前n个元素均等于val,则参考链表头删,删除直至链表第一个元素head->val != val;
- 使用while循环,依次查询链表元素,如查询到链表内 存在元素等于 val , 则将指针跳过当前指针,并free()当前指针所指链表内存,将当前元素的 前一元素存放指针指向当前元素的下一个元素。
代码段如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
typedef struct ListNode Node; //为方便写代码,将struct ListNode 重定义为 Node
struct ListNode* removeElements(struct ListNode* head, int val){
//如果head 为空,则直接返回head
if(head == NULL)
return head;
//查询head的前n个元素是否为val ,如果是,则进行头删
while(head->val == val && head != NULL)
{
Node* next = head->next;
free(head);
head = next;
//如果此 if 成立,则表明链表中所有元素均为val ,为避免野指针出现,直接返回NULL
if(head == NULL)
return NULL;
}
//经过以上代码,可以确定链表剩下的元素一定不以 val 开头,定义两个Node* 指针,
//prve用以记录元素的上一个元素地址,用以链表链接
//curr用以记录当前元素的地址,用以遍历
Node* prve = NULL;
Node* curr = head;
//循环直至curr == NULL ,表示链表遍历完成
while(curr)
{
//初次进入while循环 此 if 一定不成立(在进入此循环前已删除开头的val),进入else语句进行迭代
if(curr->val == val)
{
//迭代至某次,找到为val 的数据,则将prve 指向curr的下一元素,将链表跳过当前元素链起
prve->next = curr->next;
//free当前地址的内存
free(curr);
curr = prve->next;
}
//
else
{
prve = curr;
curr = curr->next;
}
}
return head;
}
请批评指正。