Leetcode: 203. 移除链表元素

题目

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

难度：简单

题目链接：203. 移除链表元素

示例 1：

复制代码

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

示例 2：

复制代码

输入：head = [], val = 1
输出：[]

示例 3：

复制代码

输入：head = [7,7,7,7], val = 7
输出：[]

方法一：

题目解析：

遍历链表，删除指定元素(val)

代码展示

cpp 复制代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */


struct ListNode* removeElements(struct ListNode* head, int val){
    struct ListNode* pre = NULL,*cur = head;
    while(cur)
    {
        if(cur->val == val)
        {
            struct ListNode* next = cur->next;
            free(cur);
            if(pre == NULL) //删除第一个节点时，free之后，head就变成了也指针
            {
                //为了避免head变成野指针
                head = next;
            }
            else
            {
                pre->next = next;
            }
            cur = next;//继续去找元素
        }
        else
        {
            pre = cur;
            cur = cur->next;
        }
    }
    return head; //返回新的头节点
}

删除元素的主要思想

这样的话就要考虑如何找被删除元素的前一个结点，和被删除元素的下一个结点。

于是这里采用双指针

pre 指针记录前一个结点，cur 记录指向的当前结点

前提是：删去的不是第一个结点的情况下

第一步

第二步

删除的结点是第一个结点时：

即此时的 pre 是NULL ,cur指向的是head(第一个结点)，删去结点（free(cur)）。

这里free(cur) 会把第一个结点的内存空间释放返回给操作系统，pre->next = next;会出现问题

加上

cpp 复制代码

            if(pre == NULL) //删除第一个节点时，free之后，head就变成了也指针
            {
                //为了避免head变成野指针
                head = next;
            }
            else
            {
                pre->next = next;
            }
            cur = next;//继续去找元素

if(pre == NULL) --- (只有一个结点的情况而且是要删除的结点)

这种情况直接让 head 指向 next(代表cur->next,这里也就是NULL)

因为要删除的元素可能不仅仅是一个，也有可能是多个。所以每次删完一个元素的时候，注意pre指针的指向和cur指针的指向

之后再去重复上述第二步，直到cur遍历完链表。

方法二

主要思想：

先让cur指向第一个结点，遍历结点，把不等于val的值给放到新的链表里。然后返回新的链表头指针即可。

代码

cpp 复制代码

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

图解