【Java实现】移除链表元素

🎈目录🎈

问题描述🔒

输入输出实例：

解题分析🔑

代码实现🔓

---

题目入口📌：移除链表元素

问题描述

     给你一个链表的头节点 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;
    }
}