原题链接
难度: e a s y \color{Green}{easy} easy
题目描述
给你一个链表的头节点 h e a d head head 和一个整数 v a l val val ,请你删除链表中所有满足 N o d e . v a l = = v a l Node.val == 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 , 1 0 4 ] [0, 10^{4}] [0,104] 内
- 1 < = N o d e . v a l < = 50 1 <= Node.val <= 50 1<=Node.val<=50
- 0 < = v a l < = 50 0 <= val <= 50 0<=val<=50
算法
(迭代)
删除结点需要考虑以下问题
- 找到该结点的前一个结点
- 进行删除操作
所以针对头结点,我们需要创建一个指向头结点的指针 auto dummy = new ListNode(0, head);
,后面先判断当前结点是否存在?在存在的的基础上判断改结点的数值是否等于 val
如果等于就接着往后走,如果不等,就把该结点接在我们创建的虚拟结点的下一个位置。
复杂度分析
-
时间复杂度: O ( n ) O(n) O(n),其中 n n n 是链表的长度。需要遍历链表一次
-
空间复杂度 : O ( 1 ) O(1) 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 ) O(n) O(n),其中 n n n 是链表的长度。递归过程中需要遍历链表一次。
-
空间复杂度 : O ( n ) O(n) O(n),其中 n n n 是链表的长度。空间复杂度主要取决于递归调用栈,最多不会超过 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;
}
};