题目入口📌:移除链表元素
问题描述
给你一个链表的头节点
head
和一个整数val
,请你删除链表中所有满足Node.val == val
的节点,并返回 新的头节点 。输入输出实例:
解题分析
本题是要删除所有跟 val 值相等的结点,我们可以分为两步,先查找,再删除。我们可以遍历多次记录符合条件的结点位置,再遍历一遍删除。但是这种解法在面试时并不是面试官心中的最优解,为了让面试官对我们刮目相看,我们要在一次遍历中完成。
对于单向链表来说,查找结点很简单,但是删除结点有点难。如下图,比如我找到 ‘34’ 这个结点,但是我想要删除就要知道 ‘23’ 这个结点,只有修改了 ‘23’ 中next中的地址,我们才能删除 ‘45’ 这个结点。
为了在一次遍历中就删除完全部满足条件的结点,我们可以使用双指针形式,一个用于查找(fast),一个用于删除(slow),且fast在slow前面,每次遍历都走一步。
ListNode slow = head; ListNode fast = head.next;
当 fast 找到符合结点的时候,就修改slow结点中next 进行修改,如果 fast 没有找到符合的结点,那fast和slow一直向后走,直到fast走到最后一个结点。
while(fast != null){ if(fast.val == val){ slow.next = fast.next; fast = fast.next; }else{ slow = slow.next; fast = fast.next; }
之后,我们还忘记考虑一个特殊情况,我们忘记判断头结点是否符合条件
if(head.val == val){ head = head.next;
那么,我想问一下大家,这段代码应该放到什么位置,是 位置1 还是 位置2 ?
答案是位置2,如果我们放到 位置1 就是出现链表 头结点满足条件 而未被删去的情况,如下图
这样,就会导致有漏网之鱼,所以我们应该把 头结点检测放到最后面
(注:图示中的代码漏掉一段,仅为便于理解,正确代码在最下方)
代码实现
class Solution { public ListNode removeElements(ListNode head, int val) { if(head == null) return head;//判断链表是否为空 ListNode slow = head; ListNode fast = head.next; while(fast != null){ if(fast.val == val){ slow.next = fast.next; fast = fast.next; }else{ slow = slow.next; fast = fast.next; } } if(head.val == val){ head = head.next; } return head; } }