本题来源于leetcode
1.题目描述:
这里是原题给的提示:
这里是原题给的接口函数与链表定义形式:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* removeElements(struct ListNode* head, int val){
}
下面开始手撕这道题:
2.思路:
2.1 大致框架
1.定义两个指针一个prev
一个cur
2.遍历一遍链表,然后当发现为我所要删除的元素所在的节点,即此时cur的value等于要删除的值
3.删除cur结点,``prev->next=cur->next然后
cur=cur->next`指向下一个结点。
2.2 初步实现
如何遍历
通过while循环通过cur指空来退出循环,在循环中安排一个分支,如果符合要求就删除,如果不符合要求就继续遍历下一个
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode*cur= head;
struct ListNode*prev=NULL;
while(cur)
{
if(cur->val=val)
{
prev->next=cur->next;
free(cur);
cur=prev->next;
}
else
{
prev=cur;
cur=cur->next;
}
}
}
然而此时发现对于next容易不清晰,为了简洁明了,用3个指针
prev
cur
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;
prev->next=next;
free(cur);
cur=next;
}
else
{
prev=cur;
cur=cur->next;
}
}
}
这样就比较清楚
2.3 特殊情况需要分类讨论
- 第一个就碰到了存放value的节点,这样的情况下,假如测试样例为上图所示,那么原来的头节点
head
节点就不是1而是2了,返回值head
这里要处理,同时prev是NULL(需要解决)
以下是解决方法:在判断元素是否等于value中再加一个判断语句
if(cur->val==val)
{
struct ListNode*next=cur->next;
if(prev==NULL)//即该点cur为头
{
free(cur);
head=next;
cur=next;
}
else
{
prev->next=next;
free(cur);
cur=next;
}
}
- 连续节点要删除(本来就可以解决)
- 最后一个节点要删除(本来也可以解决)
- 全是6(也能解决)
3.整体实现
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)//即该点cur为头
{
free(cur);
head=next;
cur=next;
}
else
{
prev->next=next;
free(cur);
cur=next;
}
}
else
{
prev=cur;
cur=cur->next;
}
}
return head;
}
总结:
这道题的主要要素还是一个分类讨论
4.哨兵位头节点法
设置一个哨兵头节点来防止分类讨论问题,这个哨兵一般是开辟一个节点来存储head的地址,使head变成非头节点
struct ListNode* removeElements(struct ListNode* head, int val) {
struct ListNode* guardHead = (struct ListNode*)malloc(sizeof(struct ListNode));
guardHead->next = head;
struct ListNode* prev = guardHead;
struct ListNode* cur = head;
while (cur)
{
if (cur->val == val)
{
struct ListNode* next = cur->next;
prev->next = next;
free(cur);
cur = next;
}
else
{
prev = cur;
cur = cur->next;
}
}
head = guardHead->next;//防止内存泄漏,不要直接return哨兵,转到head比较好
free(guardHead);
return head;
}
小结:
这题没必要哨兵,因为没有显得更简单,当然两种方法都是可以的。
这也是一般遇到这种类型题目的两种思路