代码随想录刷题-链表-设计链表

夏日の blog

已于 2023-03-18 18:33:58 修改

阅读量102

点赞数

分类专栏： # 算法刷题总结文章标签：链表数据结构 leetcode

于 2023-03-13 23:11:03 首次发布

本文链接：https://blog.csdn.net/zss192/article/details/129510524

版权

算法刷题总结专栏收录该内容

39 篇文章 1 订阅

订阅专栏

设计链表的实现。您可以选择使用单链表或双链表。单链表中的节点应该具有两个属性：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

我的解法

题意很直接，就是链表的常见操作。但刚开始写的时候不太明白 class 里面怎么放头节点，看了代码随想录部分题解才明白在 class 里放一个节点的 struct 就可以。同样这题在头节点前加上一个虚拟头节点会更方便点。

注意下插入节点赋值 next 时的先后顺序，如1 3中间插入2，先让节点2的 next 等于节点1的 next，再让节点1的 next 等于节点2。如果顺序反了的话，先让节点1的 next 等于节点2，这样节点2的 next 就找不到节点3了。

class MyLinkedList {
   public:
    struct Node {
        int val;
        Node* next;
        Node(int data) : val(data), next(nullptr) {}
    };
    // 初始化虚拟头节点，设置长度为0
    MyLinkedList() {
        fakeNode = new Node(0);
        size = 0;
    }
    
    int get(int index) {
        // 虽然通过了LeetCode，但这里没判断index<0的情况，要加上
        if (index + 1 > size) {
            return -1;
        }
        Node* tmp = fakeNode;
        for (int i = 0; i < index + 1; i++) {
            tmp = tmp->next;
        }
        // 遍历后tmp就是index位置的节点
        return tmp->val;
    }

    void addAtHead(int val) {
        Node* newNode = new Node(val);
        // 必须先设置newNode的next，否则就找不到真正的头节点了
        newNode->next = fakeNode->next;
        fakeNode->next = newNode;
        size++;
    }

    void addAtTail(int val) {
        Node* tmp = fakeNode;
        Node* newNode = new Node(val);
        for (int i = 0; i < size; i++) {
            tmp = tmp->next;
        }
        tmp->next = newNode;
        size++;
    }

    void addAtIndex(int index, int val) {
        if (index > size) {
            return;
        }
        if (index < 0) {
            addAtHead(val);
            return;
        }
        Node* newNode = new Node(val);
        Node* tmp = fakeNode;
        for (int i = 0; i < index; i++) {
            tmp = tmp->next;
        }
        // 遍历后tmp位于index的前一个位置
        newNode->next = tmp->next;
        tmp->next = newNode;
        size++;
    }

    void deleteAtIndex(int index) {
        // 还是没判断index<0的情况，要加上
        if (index + 1 > size) {
            return;
        }
        Node* tmp = fakeNode;
        for (int i = 0; i < index; i++) {
            tmp = tmp->next;
        }
        // 遍历后tmp位于index的前一个位置
        Node* deleteNode = tmp->next;
        tmp->next = tmp->next->next;
        delete deleteNode;  // 手动删除节点内存
        size--;
    }

    void printNode() {
        Node* tmp = fakeNode;
        for (int i = 0; i < size; i++) {
            tmp = tmp->next;
            cout << tmp->val;
        }
    }

   private:
    Node* fakeNode;
    int size;
};

时间复杂度分析
- get(int index)：时间复杂度为 O(n)，因为需要遍历整个链表，找到 index 位置的节点。
- addAtHead(int val)：时间复杂度为 O(1)，因为只需要修改头节点的指向，不需要遍历整个链表。
- addAtTail(int val)：时间复杂度为 O(n)，因为需要遍历整个链表，找到尾节点，并在其后面添加新节点。
- addAtIndex(int index, int val)：时间复杂度为 O(n)，因为需要遍历整个链表，找到 index 位置的前一个节点，并在其后面添加新节点。
- deleteAtIndex(int index)：时间复杂度为O(n)，因为需要遍历整个链表，找到index位置的前一个节点，并删除其后面的节点。
空间复杂度：O(n)。这段代码中使用了一个结构体Node来表示单链表的节点，以及一个指向虚拟头节点的指针fakeNode和一个记录链表长度的变量size。其中，节点的数量和虚拟头节点的数量都是n+1个，因此空间复杂度是O(n)。

需要注意的点如下：

注意 index 的极端情况，本题我就没判断 index<0的情况，虽然通过了测试用例，但严格意义上并不全对
代码中的 size 和 fakeNode 前面加个 _ 更好，如 _size ，这样能和其他变量区分，当然也可以用 this->size

代码随想录代码

和我的代码类似，我用的 for 循环遍历，他用的 while 遍历。注释已经很清晰了，就不过多讲解了。

class MyLinkedList {
public:
    // 定义链表节点结构体
    struct LinkedNode {
        int val;
        LinkedNode* next;
        LinkedNode(int val):val(val), next(nullptr){}
    };

    // 初始化链表
    MyLinkedList() {
        _dummyHead = new LinkedNode(0); // 这里定义的头结点 是一个虚拟头结点，而不是真正的链表头结点
        _size = 0;
    }

    // 获取到第index个节点数值，如果index是非法数值直接返回-1， 注意index是从0开始的，第0个节点就是头结点
    int get(int index) {
        if (index > (_size - 1) || index < 0) {
            return -1;
        }
        LinkedNode* cur = _dummyHead->next;
        while(index--){ // 如果--index 就会陷入死循环
            cur = cur->next;
        }
        return cur->val;
    }

    // 在链表最前面插入一个节点，插入完成后，新插入的节点为链表的新的头结点
    void addAtHead(int val) {
        LinkedNode* newNode = new LinkedNode(val);
        newNode->next = _dummyHead->next;
        _dummyHead->next = newNode;
        _size++;
    }

    // 在链表最后面添加一个节点
    void addAtTail(int val) {
        LinkedNode* newNode = new LinkedNode(val);
        LinkedNode* cur = _dummyHead;
        while(cur->next != nullptr){
            cur = cur->next;
        }
        cur->next = newNode;
        _size++;
    }

    // 在第index个节点之前插入一个新节点，例如index为0，那么新插入的节点为链表的新头节点。
    // 如果index 等于链表的长度，则说明是新插入的节点为链表的尾结点
    // 如果index大于链表的长度，则返回空
    // 如果index小于0，则在头部插入节点
    void addAtIndex(int index, int val) {

        if(index > _size) return;
        if(index < 0) index = 0;        
        LinkedNode* newNode = new LinkedNode(val);
        LinkedNode* cur = _dummyHead;
        while(index--) {
            cur = cur->next;
        }
        newNode->next = cur->next;
        cur->next = newNode;
        _size++;
    }

    // 删除第index个节点，如果index 大于等于链表的长度，直接return，注意index是从0开始的
    void deleteAtIndex(int index) {
        if (index >= _size || index < 0) {
            return;
        }
        LinkedNode* cur = _dummyHead;
        while(index--) {
            cur = cur ->next;
        }
        LinkedNode* tmp = cur->next;
        cur->next = cur->next->next;
        delete tmp;
        _size--;
    }

    // 打印链表
    void printLinkedList() {
        LinkedNode* cur = _dummyHead;
        while (cur->next != nullptr) {
            cout << cur->next->val << " ";
            cur = cur->next;
        }
        cout << endl;
    }
private:
    int _size;
    LinkedNode* _dummyHead;

};