【每日一道算法题】移除链表元素
给你一个链表的头节点 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 输出:[]
思路
如何实现单链表的删除操作?
我们知道,单链表的每个节点存着两个变量,一个变量val是数据值,一个变量next是下一个节点的地址值。如果想删除链表当中的某个节点,比如下图当中的0x002,就要把0x003赋值给0x001的next。但此时你会发现,0x002依然指向的是0x003,这样不是还存在内存当中吗?是的,但是通过java自己的内存回收机制,会将其占用的内存自动释放。
因此由于单链表的特殊性,链表的节点都是通过它的前一个节点来移除的,即让当前节点的next指针直接指向下下个(后两个)节点,散落的节点(如下图0x002)会通过内存回收机制自动删除。
记录我的代码
做了一个小时,做不出来,放弃了。
后来下午看了题解,画了十三分钟终于做了出来。第一次做大体思路是正确的,但是一来不清楚当前节点的下下个节点的地址值可以用Node.next.next
来表达,二来没有设置虚拟头节点,单独写了一段逻辑来判断头节点的数据值是否等于目标值,做的过程非常混乱。
class Solution {
public ListNode removeElements(ListNode head, int val) {
//定义一个虚拟头节点dummyhead
ListNode dummyhead = new ListNode(0);
dummyhead.next = head;
//移除链表元素,如果下一个节点的值等于目标值,则删除下一个节点,循环条件是下一个节点不为空
//定义当前节点curNode为虚拟头节点
ListNode curNode = dummyhead;
while(curNode.next != null){
//下一个节点不为空
//如果下一个节点值等于val
if(curNode.next.val == val){
//删除下一个链表节点
curNode.next = curNode.next.next;
}else{//这里为什么用else:如果下一个节点要删除,那么当前节点指向的就是下下个节点,符合我们的需求;如果下一个节点不用删除,那么当前节点需要移动到下一个节点才能判断下下个节点的数据是否等于val。
//如果下一个节点值不等于val,那么当前节点需要变成下一个节点
curNode = curNode.next;
}
}
//返回头节点
return dummyhead.next;
}
}
力扣官方题解
递归
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;
}
}
迭代
由于头结点(首元节点)可能需要被删除,所以设置一个虚拟的头节点。
class Solution {
public ListNode removeElements(ListNode head, int val) {
ListNode dummyHead = new ListNode(0);
dummyHead.next = head;
ListNode temp = dummyHead;
while (temp.next != null) {
if (temp.next.val == val) {
temp.next = temp.next.next;
} else {
temp = temp.next;
}
}
return dummyHead.next;
}
}
参考学习资料:代码随想录