思路一:
类似双指针,prev在后,cur在前先行,如果cur->val==val,则prev ->next = cur->next,并free掉cur.然后将prev->next赋值给cur,继续一前一后的向下走,直到cur==NULL。
其中有特殊情况,如果第一个cur->val==val,需要用头删的处理方式,改变头部指针。
但是在单链表中,曾讲解到,如果要改变实参,必须要接收实参的地址,但这里虽然改变了头部指针,但接收的依然是实参的一份临时拷贝,也就是说,如果调用完函数之后,并不会改变实参。实参确实没被改变,但是我返回去了一个地址,那么后续链表的访问都是以我返回的地址作为头部指针了。相当于该变了实参。
struct ListNode* removeElements(struct ListNode* head, int val)
{
struct ListNode* cur = head;
struct ListNode* prev=NULL;
while(cur)
{
if(cur->val==val)//删除
{
if(cur==head)//头删处理
{
head = cur->next;
free(cur);
cur = head;
}
else
{
prev->next = cur->next;
free(cur);
cur = prev->next;
}
}
else//向后走
{
prev = cur;
cur = cur->next;
}
}
return head;
}
思路2:将不是 val 的元素尾插到一个新的链表中。
情况一:正常尾插,把不是val的元素跳过。
情况二:全都是val,则返回一个空指针
代码思路:
用tail遍历新链表,cur遍历旧链表,将cur赋值给tail,然后cur先向后遍历,如果cur->val==val,需要跳过这个节点,这里可以创造一个新的节点,来释放节点,防止内存泄漏,将cur赋值给del,再把del->next继续给cur,顺便释放掉del,然后cur就可以把自己赋值给tail,cur继续往后走直到结束。
代码:
struct ListNode* removeElements(struct ListNode* head, int val)
{
struct ListNode* cur = head;
struct ListNode* tail= NULL;
head=NULL;
while(cur)
{
if(cur->val==val)//删除
{
struct ListNode* del = cur;//创建新节点
cur=cur->next;
free(del);
}
else//插入新链表中
{
if(tail==NULL)//第一次插入
{
head=tail=cur;
}
else//后续正常插入
{
tail->next=cur;
tail = tail->next;
}
cur = cur->next;
}
}
if(tail)//尾部置空
{
tail->next=NULL;
}
return head;
}