203. 移除链表元素

203. 移除链表元素

题目描述

删除链表中等于给定值 val 的所有节点。
​
示例:
​
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

题目解析

解法一: 使用空表头

使用空表头, 就可以不用判断单个节点的情况或多个相同节点的情况.

// 使用空表头
    public ListNode removeElementsDummy(ListNode head, int val) {
        ListNode dummy = new ListNode(-1);
        dummy.next = head;
        ListNode p = dummy;
        while (p.next != null) {
            if (p.next.val == val) {
                p.next = p.next.next;
            } else {
                p = p.next;
            }
            
        }
        return dummy.next;
    }

解法二: 不使用空表头

不使用空表头, 需要避免的有两种情况:

• 只有一个表头

• 有多个节点, 但是全是相同节点.

// 不使用空表头, 单独判断表头
    public ListNode removeElementsNoDummy(ListNode head, int val) {
        // 删除表头后还有相同表头的情况
        while (head != null && head.val == val) {
            head = head.next;
        }
        if (head == null) return null;
        ListNode p = head;
        
        while (p.next != null) {
            if (p.next.val == val) p.next = p.next.next;
            else p = p.next;
        }
        return head;
    }