删除链表中等于给定值 val 的所有节点。
题目源于领扣:https://leetcode-cn.com/problems/remove-linked-list-elements/description/
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5
分析:
- 当链表为空时,返回NULL
- 当链表不为空时
1.当前被删除节点是头节点(包含只有一个节点的情况)
2.被删除节点不是头节点
删除元素过程(以C++代码为例)
如图:
注意:
- 当删除的节点就是头结点时,新的头结点(即
cur->next
)就是NULL,满足删除指定val的所有元素的条件。 - 当判断完结点是否为头结点时,不能漏掉
cur=cur->next
这一句,因为指定val值的元素不止一个,所以在删除一个结点之后,还要让cur
继续向后走,继续排查。 - 下面的C++代码和C语言代码有一点点细微的差别,C语言中有一句
free(cur)
,但C++代码中没有,需要注意是,C语言中的链表结点是一个结构体,是由malloc开辟出来的空间,所以使用free来释放这段内存空间,是真正意义的将结点删除。而在C++代码中,涉及到类和对象,使用free释放内存仅仅释放了对象的内存,而没有调用析构函数,正确的删除对象需要delete,所以这里C++代码中不用free,删除节点实质上是将目标结点覆盖。
完整代码:
C++:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode* cur = head;
ListNode* prev = NULL;
//当链表为空
if(head==NULL)
{
return head;
}
//当链表不为空
while(cur)
{
if(cur->val == val)
{
//当前节点是头节点,那么prev==NULL
if(prev == NULL)
{
head=cur->next;
}
else{
prev->next=cur->next;
}
cur=cur->next;
}
else
{
prev=cur;
cur=cur->next;
}
}
return head;
}
};
C:
/**
- Definition for singly-linked list.
- struct ListNode {
- int val;
- struct ListNode *next;
- };
*/
struct ListNode* removeElements(struct ListNode* head, int val) {
struct ListNode* cur=head;
struct ListNode* prev=NULL;
while(cur)
{
if(cur->val==val)
{
struct ListNode* next=cur->next;
if(prev==NULL)
{
head=cur->next;
}
else
{
prev->next=cur->next;
}
free(cur);
cur = next;
}
else
{
prev=cur;
cur=cur->next;
}
}
return head;
}