203. 移除链表元素

目录

题目描述

思路分析

代码实现

复杂度

---

题目描述

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

示例 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
输出：[]
 

提示：

列表中的节点在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= k <= 50

思路分析

• 方法一：哨兵结点

如果要删除某一个结点，首先要知道该结点的前驱结点。在此题中，如果要删除的结点位于中间位置，那么非常简单；如果链表的开头有一个或者多个要删除的结点，就必须要设置一个哨兵结点，否则无法删除第一个结点。哨兵结点广泛应用于树和链表中，如伪头、伪尾、标记等，它们是纯功能的，通常不保存任何数据，其主要目的是使链表标准化，如使链表永不为空、永不无头、简化插入和删除。

代码实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode *sentinel = new ListNode(0);
        sentinel->next = head;
        ListNode *pre = sentinel, *cur = head;
        while(cur){
            if(cur->val == val){
                pre->next = cur->next;
                cur = cur->next;                
            }
            else{
                pre = cur;
                cur = cur->next;
            }
        }
        ListNode *res = sentinel->next;
        return res;
    }
};

复杂度

时间复杂度O(N)

空间复杂度O(1)