(一)问题描述
leetcode题目链接. - 备战技术面试?力扣提供海量技术面试资源,帮助你高效提升编程技能,轻松拿下世界 IT 名企 Dream Offer。https://leetcode.cn/problems/remove-linked-list-elements/description/ 给你一个链表的头节点 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
(二) 解决思路
链表包含两个部分:值和指向下一个节点的指针。假设要删除的节点是A,删除A的方式就是让A的上一个节点直接指向A的下一个节点(Java和Python会自动处理A的内存,但是C需要单独释放内存)。
这里唯一要注意的一个问题就是:要删除第一个节点该怎么办,因为第一个节点没有上一个节点。这里的解决办法有两个,一是创建一个虚拟的头节点,这样从第一个节点开始,每个节点的处理方式都是一样的,返回的时候返回虚拟头节点的下一个节点就可以了;二是在原链表的基础上进行处理,最后再单独判断一下头节点,如果头节点是要删除的节点,直接返回头节点的下一个节点就可以了。
1. 使用虚拟头节点
public ListNode removeElements(ListNode head, int val) {
// 设置一个虚拟的头节点
ListNode dummy = new ListNode();
dummy.next = head;
ListNode cur = dummy;
while (cur.next != null) {
if (cur.next.val == val) {
cur.next = cur.next.next;
} else {
cur = cur.next;
}
}
return dummy.next;
}
2. 原地修改
public ListNode removeElements(ListNode head, int val) {
while(head!=null && head.val==val) {
head = head.next;
}
ListNode curr = head;
while(curr!=null && curr.next !=null) {
if(curr.next.val == val){
curr.next = curr.next.next;
} else {
curr = curr.next;
}
}
return head;
}
(三)易错点
- 循环和条件判断需要判断head/curr是否为空