LeetCode 203. 移除链表元素

LeetCode 203. 移除链表元素

难度：$easy$

题目描述

给你一个链表的头节点 $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,10^4]$ 内
• $1<=Node.val<=50$
• $0<=val<=50$

---

算法

(迭代)

删除结点需要考虑以下问题

• 找到该结点的前一个结点
• 进行删除操作

所以针对头结点，我们需要创建一个指向头结点的指针 auto dummy = new ListNode(0, head);，后面先判断当前结点是否存在？在存在的的基础上判断改结点的数值是否等于 val 如果等于就接着往后走，如果不等，就把该结点接在我们创建的虚拟结点的下一个位置。

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是链表的长度。需要遍历链表一次

• 空间复杂度 : $O(1)$

C++ 代码

/**
 * 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) {
        auto dummy = new ListNode(0, head);

        for (auto p = dummy; p; p = p->next) {
            auto q = p->next;
            while (q && q->val == val) q = q->next;
            p->next = q;
        }

        return dummy->next;
    }
};

---

算法2

(递归)

• 对于给定的链表，首先对除了头节点 head 以外的节点进行删除操作，然后判断 head 的节点值是否等于给定 val。如果 head 的节点值等于 val，则 head 需要被删除，因此删除操作后的头节点为 head->next；如果 head的节点值不等于 val，则 head 保留，因此删除操作后的头节点还是 head。上述过程是一个递归的过程。

• 递归的终止条件是 head 为空，此时直接返回 head。当 head 不为空时，递归地进行删除操作，然后判断 head 的节点值是否等于 val 并决定是否要删除 head。

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是链表的长度。递归过程中需要遍历链表一次。

• 空间复杂度 : $O(n)$，其中 $n$ 是链表的长度。空间复杂度主要取决于递归调用栈，最多不会超过 $n$ 层。

C++ 代码

/**
 * 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) {
        if (head == NULL) return head;

        head->next = removeElements(head->next, val);
        return head->val == val ? head->next : head;
    }
};

---

C++ 代码

/**
 * 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) {
        auto dummy = new ListNode(0, head);
        auto cur = dummy;

        while (head) {
            if (head->val == val) {
                if (head->next == nullptr) {
                    cur = cur->next = nullptr;
                    break;
                }
                head = head->next;
            } else {
                cur =  cur->next = head;
                head = head->next;
            }
        }

        return dummy->next;
    }
};