移除链表元素
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, 10^4]
内 1 <= Node.val <= 50
0 <= val <= 50
递归
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;
}
}
- 时间复杂度:O(n),其中 n 是链表的长度。递归过程中需要遍历链表一次。
- 空间复杂度:O(n),其中 n 是链表的长度。空间复杂度主要取决于递归调用栈,最多不会超过 n 层。
快慢指针
快指针无条件前进,慢指针有条件前进到快指针。检查快指针指向的元素。
class Solution {
public ListNode removeElements(ListNode head, int val) {
// 移除链表开头连续的目标值节点
while (null != head && head.val == val) {
head = head.next; // 更新头节点
}
// 初始化当前节点和前一个节点,均为头节点
ListNode cur = head;
// pre 相当于构造链表的指针
ListNode pre = head;
// 遍历链表,直到当前节点为空。根据快指针的值确定更新慢指针的next还是前进慢指针到快指针
while (cur != null) {
// 如果当前节点的值等于目标值
if (cur.val == val) {
// 前一个节点的next指向当前节点的next,相当于跳过了当前节点。重点。
pre.next = cur.next;
} else {
// 否则,前一个节点移动到当前节点
pre = cur;
}
// 当前节点移动到下一个节点
cur = cur.next;
}
// 返回处理后的链表头节点
return head;
}
}