力扣网 203. 移除链表元素
题目描述
给你一个链表的头节点 head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点 。
思路分析
针对如图的普通情况,不能简单的遍历到对应位置然后进行释放,一方面会丢掉下一个结点的地址,同时上一个结点的成员指针也会变成野指针。基本的方法是,每次循环,都保留该节点的下一个结点地址,也要保留上一个结点的地址(该指针最开始指向NULL),如果该节点的值是需要删除值,让前一个结点的指针指向该节点的下一个结点后再进行释放,如果不是就把当前结点保留,继续遍历下去,
另外一种特殊情况是当头结点就是需要删除的值,如果直接让前一个结点的指针指向该节点的下一个结点,那就是对空指针的非法引用了,是不允许的,所以这里还要再加一个判断条件,即当前一个结点(设为pre)不为空时,让前一个结点的指针指向该节点的下一个结点,否则,直接将下一个结点赋给头结点,因为释放完后头结点也为野指针,此时需要进行赋值。
代码语言:javascript复制struct ListNode* removeElements(struct ListNode* head, int val) {
struct ListNode* pre=NULL;//前驱指针,最开始为NULL
struct ListNode* cur=head;//对应指针
while(cur)
{
if(cur->val==val)
{
struct ListNode* next=cur->next;//用来保存下一个结点
free(cur);
if(pre)//如果pre不为空,直接让pre-》next指向next即可 {
pre->next=next;
}
else
{
head=next;//为空则将头结点指向next
}
cur=next;
}
else//不是对应值,将当前指针赋给pre成为下一个指针的前驱指针,cur往下遍历
{
pre=cur;
cur=cur->next;
}
}
return head;//返回头结点
}