C语言——移除链表元素

一、题目简述

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
例如：

二、题目分析

在链表中，删除一个元素可能会出现多种情况，每一种都要考虑周到。
1：当删除的节点在中间时，我们除了要将该节点free掉，还要将该节点的前后节点连接起来，保证链表的完整。但在单链表中，我们只能通过该节点找到下一个节点，而不能找到它的前一个节点。所以在遍历链表的过程中，我们要用另一个指针来记录前一个节点。
2：当要删除的节点就是头节点时，我们就要改变原链表的头节点，让head=head->next，但此时前一个节点pre为NULL指针，没有必要对其进行改变，头节点newhead的前一个节点仍为空。
3：当cur等于NULL时，遍历结束。返回新的头节点newhead

三、代码分析

    struct ListNode* pre=NULL;//记录链表前一个节点
    struct ListNode* cur=head;
    struct ListNode* newHead=head;//记录新的头节点，删除时有可能删除头节点，所以可能要改变头节点。

pre记录cur的前一个节点，开始为NULL。
cur用来遍历链表。
newhead用来记录新的头节点。

1、删除头节点

        if(cur->val==val)//节点值等于要删除的值
        {
            if(cur==newHead)//删除的是头节点
            {
                newHead=cur->next;
                free(cur);
                cur=newHead;
            }
            else            //删除的不是头节点
            {
            pre->next=cur->next;
            free(cur);
            cur=pre->next;
            }
        }

找到cur->val==val时，要判断cur是不是头节点：
如果是头节点，先将newhead置为cur->next，再将cur所在的头节点free掉。pre不用变化，因为此时还在头节点前面，仍应为NULL。

如上图所示：当要删除6节点时，此时6节点正好为头节点，此时应先将newhead置为cue->next，若先free掉cur便找不到cur->next节点。pre此时为NULL，不用对其进行改变。

2、删除非头节点

如果不是头节点，pre->next要和cur->next链接起来，将cur节点free掉后，cur继续呆在pre->next节点，pre不用变化。

如上图所示:要删除6节点，要先将pre和cur->next链接起来，如果先free掉cur，那么便找不到cur->next这个节点，无法链接。free掉cur后，cur再跑到pre->next即可。此时pre->next应为5节点，而不是6节点。

3、不删除节点`

        else              //节点值不等于要删除的值
        {
            pre=cur;      pre记录cur的值
            cur=cur->next;
        }

直接让pre记录cur的位置，cur向前走一步，cur=cur->next即可。
cur为NULL时循环结束，遍历完全链表。结束后返回新的头节点newhead即可。

四、完整代码

struct ListNode* removeElements(struct ListNode* head, int val) {
    struct ListNode* pre=NULL;//记录链表前一个节点
    struct ListNode* cur=head;
    struct ListNode* newHead=head;//记录新的头节点，删除时有可能删除头节点，所以可能要改变头节点。

    while(cur)
    {
        if(cur->val==val)//节点值等于要删除的值
        {
            if(cur==newHead)//删除的是头节点
            {
                newHead=cur->next;//改变头节点
                free(cur);
                cur=newHead;//cur到新的头节点
            }
            else            //删除的不是头节点
            {
            pre->next=cur->next;//链接cur前一节点和后一节点
            free(cur);
            cur=pre->next;
            }
        }
        else              //节点值不等于要删除的值
        {
            pre=cur;      //pre记录cur的值
            cur=cur->next;//cur向前走一步，来到下一个节点
        }

    }
    
    return newHead;
}