package com.zmz.algorithm.listnode;

/**
 * @author 张明泽
 * Create by 2022/5/27 9:46
 * 移除链表元素
 * LeetCode-203
 */
public class removeListNode {
    public static void main(String[] args) {
        ListNode head = new ListNode(6);
        ListNode node2 = new ListNode(6);
        ListNode node3 = new ListNode(3);
        ListNode node4 = new ListNode(6);
        head.next = node2;
        node2.next = node3;
        node3.next = node4;
        ListNode node = removeElements(head,6);
        traverse(node);
    }

    /**
     * 设置虚拟节点
     */
    public static ListNode removeElements(ListNode head, int val) {
        ListNode header = new ListNode(-1);
        header.next = head;
        ListNode cur = header;
        while (cur.next != null) {
            if (cur.next.val == val) {
                cur.next = cur.next.next;
            } else {
                cur = cur.next;
            }
        }
        return header.next;
    }

    /**
     * 不设置虚拟节点
     */
    public static ListNode removeElements2(ListNode head, int val) {
        ListNode cur = head;
        while (cur.next != null) {
            if (cur.next.val == val) {
                cur.next = cur.next.next;
            } else {
                cur = cur.next;
            }
        }
        /**
         * 最后处理首节点
         */
        if (head.val == val) {
            head = head.next;
        }
        return head;
    }
    public static class ListNode {
        int val;
        ListNode next;
        ListNode() {}
        ListNode(int val) { this.val = val; }
        ListNode(int val, ListNode next) { this.val = val; this.next = next; }
    }
    public static void traverse(ListNode listNode) {
        while (listNode != null) {
            System.out.println(listNode.val);
            listNode = listNode.next;
        }
    }
}