LeetCode 203. Remove Linked List Elements (移除链表元素)

题目

链接

https://leetcode-cn.com/problems/remove-linked-list-elements/

问题描述

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

示例

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

提示

列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50

思路

首先在头节点之前设置一个节点，避免链表全被删除。

之后如果遇到要删除的，就将下一个结点和前一个节点连起来，不然的话就正常向下。

复杂度分析

时间复杂度 O(n)
空间复杂度 O(1)

代码

Java

    public ListNode removeElements(ListNode head, int val) {
        ListNode pre = new ListNode();
        pre.next = head;
        ListNode p = pre;
        while (head != null) {
            if (head.val == val) {
                p.next = head.next;
            } else {
                p = p.next;
            }
		head = head.next;
        }
        return pre.next;
    }