LeetCode刷题Java——203.移除链表元素(简单)

题目

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

示例

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

输入：head = [], val = 1
输出：[]

输入：head = [7,7,7,7], val = 7
输出：[]

解决方案

简单的链表遍历，在遍历的途中做一些操作，基本上都可以用快慢指针。

/**
 * 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) {
        // 处理head为null的情况有两种方法：1.直接if判断 2.创建一个新的头指针
        // 比较之后觉得方法2比较优雅一点

        // 创建一个新的头指针，避免head为null的情况
        ListNode newHead = new ListNode();
        // 只需要将新的头指针指向head(最后是返回newHead.next)，不需要设置值
        newHead.next = head;

        // 使用快慢指针的方法遍历所有node
        ListNode fast = head;
        // 这里要注意slow一开始指的要比fast慢一个节点，不然263这样的顺序，中间的6会被跳过
        ListNode slow = newHead;
        while(fast != null) {
            if(fast.val == val) {
                fast = fast.next;
                slow.next = fast;
            } else {
                fast = fast.next;
                slow = slow.next;
            }
        }
        return newHead.next;
    }
}