LeetCode 707. Design Linked List (設計連結串列)

題目

連結

https://leetcode.cn/problems/design-linked-list/

問題描述

設計連結串列的實現。您可以選擇使用單鏈表或雙鏈表。單鏈表中的節點應該具有兩個屬性：val 和 next。val 是當前節點的值，next 是指向下一個節點的指標/引用。如果要使用雙向連結串列，則還需要一個屬性 prev 以指示連結串列中的上一個節點。假設連結串列中的所有節點都是 0-index 的。

在連結串列類中實現這些功能：

get(index)：獲取連結串列中第 index 個節點的值。如果索引無效，則返回-1。
addAtHead(val)：在連結串列的第一個元素之前新增一個值為 val 的節點。插入後，新節點將成為連結串列的第一個節點。
addAtTail(val)：將值為 val 的節點追加到連結串列的最後一個元素。
addAtIndex(index,val)：在連結串列中的第 index 個節點之前新增值為 val  的節點。如果 index 等於連結串列的長度，則該節點將附加到連結串列的末尾。如果 index 大於連結串列長度，則不會插入節點。如果index小於0，則在頭部插入節點。
deleteAtIndex(index)：如果索引 index 有效，則刪除連結串列中的第 index 個節點。

示例

MyLinkedList linkedList = new MyLinkedList();
linkedList.addAtHead(1);
linkedList.addAtTail(3);
linkedList.addAtIndex(1,2); //連結串列變為1-> 2-> 3
linkedList.get(1); //返回2
linkedList.deleteAtIndex(1); //現在連結串列是1-> 3
linkedList.get(1); //返回3

提示

所有val值都在 [1, 1000] 之內。
操作次數將在  [1, 1000] 之內。
請不要使用內建的 LinkedList 庫。

思路

正常按照連結串列的性質處理即可，這裡用的是單鏈表。

複雜度分析

時間複雜度 O(n)
空間複雜度 O(n)

程式碼

Java

 class MyLinkedList {
        int size;
        ListNode head;

        public MyLinkedList() {
            size = 0;
            head = new ListNode(0);
        }

        public int get(int index) {
            if (index < 0 || index >= size) {
                return -1;
            }
            ListNode curr = head;
            for (int i = 0; i < index + 1; i++) {
                curr = curr.next;
            }
            return curr.val;
        }

        public void addAtHead(int val) {
            addAtIndex(0, val);
        }

        public void addAtTail(int val) {
            addAtIndex(size, val);
        }

        public void addAtIndex(int index, int val) {
            if (index > size) {
                return;
            }
            if (index < 0) {
                index = 0;
            }

            size++;
            ListNode pred = head;
            for (int i = 0; i < index; i++) {
                pred = pred.next;
            }
            ListNode toAdd = new ListNode(val);
            toAdd.next = pred.next;
            pred.next = toAdd;
        }


        public void deleteAtIndex(int index) {

            if (index < 0 || index >= size) {
                return;
            }
            size--;
            ListNode pred = head;
            for (int i = 0; i < index; i++) {
                pred = pred.next;
            }
            pred.next = pred.next.next;
        }
    }