LeetCode 203. Remove Linked List Elements
LeetCode 203. Remove Linked List Elements (移除链表元素)
题目
链接
https://leetcode-cn.com/problems/remove-linked-list-elements/
问题描述
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
示例
输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]
提示
列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50
思路
首先在头节点之前设置一个节点,避免链表全被删除。
之后如果遇到要删除的,就将下一个结点和前一个节点连起来,不然的话就正常向下。
复杂度分析
时间复杂度 O(n)
空间复杂度 O(1)
代码
Java
public ListNode removeElements(ListNode head, int val) {
ListNode pre = new ListNode();
pre.next = head;
ListNode p = pre;
while (head != null) {
if (head.val == val) {
p.next = head.next;
} else {
p = p.next;
}
head = head.next;
}
return pre.next;
}