Implement LinkedList Class

 

 

 

public class Solution
{
  public static void main(String[] args)
  {
    // 1 --> 2 --> 3 --> 5 --> 4 --> 6 --> 0 --> null
    // 0     1     2     3     4     5     6
    MyLinkedList l = new MyLinkedList();

    l.addHead(5);
    l.addHead(3);
    l.addHead(2);
    l.addHead(1);
    l.addTail(4);
    l.addTail(6);
    l.addTail(0);

    ListNode head = l.head;
    while (head != null) {
      System.out.print(head.value + " -> ");
      head = head.next;
    }

    System.out.println();
    System.out.print(l.get(1) + " , ");
    System.out.print(l.get(0) + " , ");
    System.out.print(l.get(5) + " , ");
    System.out.print(l.get(6) + " , ");

    System.out.println();

    l.remove(2);
    l.remove(5);
    head = l.head;
    while (head != null) {
      System.out.print(head.value + " -> ");
      head = head.next;
    }

    System.out.println();
    l.removeVal(4);
    l.removeVal(5);

    head = l.head;
    while (head != null) {
      System.out.print(head.value + " -> ");
      head = head.next;
    }
  }
}


public class ListNode {
    int value;
    ListNode next;
    // ListNode prev;
    public ListNode(int value) {
        this.value = value;
    }
}

public class MyLinkedList {
    public ListNode head;
    public ListNode tail;
    public int length;
    
    public MyLinkedList() {
        head = null;
        tail = null;
        length = 0;
    }

    public void addHead(int value) {
        ListNode newHead = new ListNode(value);
        newHead.next = head;
        head = newHead;
        if (length == 0) {  // records previous node number
            tail = head;
        }
        length++;
    }
    
    public void addTail(int value) {
        ListNode newTail = new ListNode(value);
　　　　 if (length == 0) {
        　　head = newTail;
　　　　　} else {
　　　　　　　tail.next = newTail; 
　　　　　　　　　 // newTail.prev = tail;
　　　　　}    
　　　　tail = newTail;
　　　　length++;
    }
    
    public Integer get(int index) {    // Zero based
        if (index < 0 || index >= length) {
            return null;
        }
        ListNode cur = head;
        //  1 → 2 → 3 → 4 → 
        //      0   1
　　　　　//  0   1   2  
　　　　　for (int i = 0; i < index; i++) {
            cur = cur.next;
        }
        return cur.value;
    }

    public void remove(int index) {
        if (index < 0 || index >= length) {    
            return;
        }
        //  1 → 2 → 3 → 4 → 
        //      0   1
　　　　　//  0   1   2  
　　　　　if (index == 0) {    
    　　　　　if (length == 1) {    // It cannot be length = 0, because index = length has been filtered out!
        　　　　　tail = null;
    　　　　　}
    　　　　　head = head.next;
　　　　　} else {
　　　　　　　ListNode prev = head;    
    　　 　　for (int i = 0; i < index - 1; i++) {
        　　　　prev = prev.next;
    　　　　　}
    　　//What if index is the position of tail node?  
    　　　　if (prev.next == tail) {    // ← Or index == length - 1
       　　　　tail = prev;
    　　　　}
    　　　　prev.next = prev.next.next;
　　　　}
　　　　length--;
    }
    
    
    //Remove first value val, not index
    public ListNode removeVal(int val) {
        if (head == null) {
            return null;
        }
        ListNode cur = head;
        if (head.value == val) {
            head = head.next;
            length--;
            return cur;
        }

        ListNode prev = null;
        while (cur != null) {
            if (cur.value == val) {
                prev.next = prev.next.next;
                cur.next = null;
                length--;
                return cur;
            }
            prev = cur;
            cur = cur.next;
        }
        return null;
    }

    public int size() {
        return length;
    }
}