LeetCode第15天--移除链表元素

移除链表元素

问题描述

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

示例 

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]
输入：head = [], val = 1
输出：[]
输入：head = [7,7,7,7], val = 7
输出：[]

解题思路：一共分为三种情况

第一种是正常情况：定义两个指针变量，cur=head头节点，prev最初指向NULL,之后当cur->val!=val时prev = cur;cue=cur->next;当cur->val == val时，prev->next = cur->next;cur= prev->next;图解如下：

 

第二种是等于val的节点有连续几个：同样也是定义两个指针变量，cur=head头节点，prev最初指向NULL,之后当cur->val!=val时prev = cur;cue=cur->next;当cur->val == val时，prev->next = cur->next;cur= prev->next;图解如下：

 前两种情况总结下来可以当成第一种普通情况来处理。

第三种情况是需要移除的点是head：我们需要先判断，当cur->val==val时，prev=NULL,这时我们需要重新定义一个next节点指向，另next=head->next;销毁掉cur,将head和cur的指向都改为next的指向。当prev！=NULL时，和前两种操作就一样了。图解如下：

 代码实现如下：

struct ListNode* removeElements(struct ListNode* head, int val)
{
    struct ListNode* cur = head;
    struct ListNode* prev = NULL;
    while(cur)
    {
        if(cur->val != val)
        {
            prev = cur;
            cur = cur->next;    
        }
        else
        {
            struct ListNode* next = cur->next;
            if(prev == NULL)
            {
                free(cur);
                head = next;
                cur = next;
            }
            else
            {
                prev->next = cur->next;
                free(cur);
                cur = prev->next; 
            }           
        }
    }
    return head;
}