【刷题日记】1.移除链表元素

目录

题目介绍

总结

---

题目链接：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题时一定要画图和考虑特殊情况