目录
题目链接:203. 移除链表元素 - 力扣(LeetCode) (leetcode-cn.com)
题目介绍
这道题的题意主要包含两方面
1.删除与指定val相同的节点
2.返回新的头节点
我们要完成这道题,我们只能从头到尾遍历一遍
我们以题目中的实例1演示
我们先想到了用一个指针来访问
当我们走到需要删除的值时,我们发现用一个指针无法完成任务
我们无法找到cur之后的节点
这时我们想到了用两个指针来遍历链表
cur指针在链表最前面,prev指针是在cur的后面,对比cur的val和val是否相等
但是当我们free掉cur之后我们又无法访问cur的next了
所以我们还需要另一个指针来解决问题
如果相等我们就free掉cur所指向的节点,然后cur指向它的下一个节点
这样一直循环直到cur到链表的尾也就是NULL
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode*cur=head;
struct ListNode*prev=NULL;
while(cur)
{
struct ListNode*next=cur->next;
if(cur->val==val)
{
prev->next=next;
free(cur);
cur=next;
}
else
{
prev=cur;
cur=cur->next;
}
}
return head;
}
似乎这样的写法是没有问题的,我们提交时就会出现问题
我们发现出现空指针的错误,原因是我们没有考虑如果整个链表都要删除的情况,这就导致了prev一直是NULL所以我们需要考虑如果整个链表都要删除的情况
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode*cur=head;
struct ListNode*prev=NULL;
while(cur)
{
struct ListNode*next=cur->next;
if(cur->val==val)
{
if(prev==NULL)
{
free(cur);
head=next;
cur=next;
}
else
{
prev->next=next;
free(cur);
cur=next;
}
}
else
{
prev=cur;
cur=cur->next;
}
}
return head;
}
这样我们就能够解决这个问题了
提交通过
总结
我们在写链表oj题时一定要画图和考虑特殊情况