给你一个链表的头节点 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 <= val <= 50
输入: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 <= val <= 50
代码如下:
class Solution {
public ListNode removeElements(ListNode head, int val) {
// 先处理头节点可能等于 val 的情况
while (head!= null && head.val == val) {
head = head.next;
}
if (head == null) {
return head;
}
ListNode p, pre;
pre = head;
p = pre.next;
while (p!= null) {
if (p.val == val) {
pre.next = p.next;
} else {
pre = p;
}
p = p.next;
}
return head;
}
}
错误代码:
class Solution {
public ListNode removeElements(ListNode head, int val) {
if(head == null) {
return head;
}
ListNode p ,pre;
pre = head;
p = pre.next;
while( p!= null) {
if(p != null && pre.val != val) {
pre = p;
} else {
pre.next = p.next;
}
p = p.next;
}
return head;
}
原因:
运行超时,
如果只判断一次 head.val == val
然后就进行后续操作,那么当链表开头有多个需要删除的节点时,程序无法正确处理。
例如,给定链表 [6,6,6,1,2,6,3]
,要删除值为 6
的节点。如果只判断一次头节点,那么当处理完第一个 6
后,新的头节点还是值为 6
的节点,但程序没有继续处理这个新的头节点。
补充:链表的基础操作:
(1)创建:
LinkedList<Integer> list = new LinkedList<>();
(2)添加元素
list.add(1);//尾插法添加元素1
list.add(2,99)//2表示索引位置,99表示添加的元素
System.out.println(list.toString());//表示输出链表
时间复杂度O(1)
(3)搜索
int index = list.indexOf(99);//查找元素99的位置,返回索引 时间复杂度O(n)
(4)更新
list.set(2,88);//将索引2的元素更新为88 时间复杂度O(n)
(5)删除
list.remove(2);//移除索引为2的元素 时间复杂度O(n)
(6) 计算链表长度
int length = list.size();
(7)访问元素
list.get(2); 时间复杂度O(1)