203. 移除链表元素

203. 移除链表元素

虚拟头节点迭代

public class Algorithm {

    public static void main(String[] args) {

        int[] arr = {1, 2, 6, 3, 4, 5, 6};
        ListNode head = new ListNode(arr);

        System.out.println(head);
        System.out.println(new Solution().removeElements(head, 6));
    }
}

class Solution {
    public ListNode removeElements(ListNode head, int val) {

        /**
         * 涉及到链表遍历，使用虚拟头节点更方便
         */
        ListNode dummyHead = new ListNode();
        dummyHead.next = head;

        ListNode prev = dummyHead;

        while (prev.next != null){
            if (prev.next.val == val){

                ListNode tem = prev.next;
                prev.next = tem.next;
                tem.next = null;
            }
            else {
                prev = prev.next;
            }
        }

        return dummyHead.next;
    }
}

/**
 * ListNode类是节点类
 * 其对象只是一个节点，而不是链表
 */
class ListNode {

    public int val;
    public ListNode next;

    public ListNode(int val, ListNode next) {

        this.val = val;
        this.next = next;
    }

    public ListNode(int val) {

        this.val = val;
    }

    public ListNode(){}

    /**
     * 在构造方法里根据一个数组来构建链表
     */
    public ListNode(int[] arr){

        if (arr == null || arr.length == 0){
            throw new IllegalArgumentException("数组是空的");
        }

        /**
         * this指这个ListNode对象
         * 将其设置为这个链表的头节点，就可以根据头节点生成链表了
         */
        this.val = arr[0];
        ListNode prev = this;

        for (int i = 1; i < arr.length; i++) {

            prev.next = new ListNode(arr[i]);
            prev = prev.next;
        }
    }

    /**
     * 当打印头节点对象时，就会打印出整个链表
     */
    @Override
    public String toString(){

        StringBuilder str = new StringBuilder();

        ListNode curr = this;
        while (curr != null){
            str.append(curr.val + "——>");
            curr = curr.next;
        }
        str.append("null");

        return str.toString();
    }
}

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

递归

class Solution {
    public ListNode removeElements(ListNode head, int val) {

        /**
         * 边界条件：如果head为空，就返回null
         */
        if (head == null){
            return null;
        }

        /**
         * 将大问题化为小问题：即将一个链表分为head和子链表，当子链表删除元素后得到一个新链表，让head指向它
         */
        head.next = removeElements(head.next, val);

        /**
         * 递归的思路就是：假想head后面的子链表操作都已经完成，现在剩下的就是比较head和子链表的头节点head.next，看谁是最终链表的头节点
         * 如果head节点需要删除，那head.next就是最终的头节点
         */
        if (head.val == val){
            return head.next;
        }
        else{

            /**
             * 否则head总会是新链表的头节点，返回head
             */
            return head;
        }

//        /**
//         * 可以简写成如下
//         */
//        if (head == null){
//            return null;
//        }
//
//        head.next = removeElements(head.next, val);
//
//        return head.val == val ? head.next : head;
    }
}

/**
 * 时间复杂度 O(n)
 * 空间复杂度 O(n)
 */

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