给你一个链表的头节点 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
输出:[]
思路
- 删除的一个或多个节点位于链表的头部时,就需要哨兵节点处理
- 定义一个哨兵节点连接链表
- 定义一个前驱节点pre指向哨兵节点,当前节点cur指向下一个节点
循环条件cur!=null:
- 如果cur.val==val就删除
- 否则pre=cur后移
- cur后移
- 返回哨兵节点的下一个节点就是新的头结点
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode removeElements(ListNode head, int val) {
ListNode dummyHead=new ListNode(-1);
dummyHead.next=head;
ListNode pre=dummyHead;
ListNode cur=head;
while(cur!=null){
if(cur.val==val){
pre.next=cur.next;
}else{
pre=cur;
}
cur=cur.next;
}
return dummyHead.next;
}
}
复杂度
时间复杂度O(n)
空间复杂度O(1)