小鑫的算法之路：leetcode0707 设计链表

题目

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

解法1：单链表

该题实际为链表的设计，针对链表需要自定义节点node，考虑到内存安全，将下一跳节点指针定义为智能指针类型，避免内存泄露。为了便于判断索引值是否有效，维护一个表示节点个数的参数，当添加节点时递增该参数，当删除节点时递减该参数。为了便于流程的处理归一，定义了一个哨兵空节点。代码如下：

class MyLinkedList {
private:
    // node类型定义，将next声明为智能指针，避免内存泄露
    struct node {
        node() : node(0, nullptr) {}

        node(int val) : node(val, nullptr) {}

        node(int val, std::unique_ptr<node>&& node) 
        {
            this->val = val;
            this->next = std::move(node);
        }

        int val;
        std::unique_ptr<node> next;
    };
    
public:
    MyLinkedList() : guard_(std::make_unique<node>()), size_(0) {}
    
    int get(int index)  // 时间复杂度为O(n)
    {
        if (!index_valid(index)) {
            return -1;
        }

        node* node = guard_.get();
        for (int i = 0; i <= index; ++i) {
            node = node->next.get();
        }

        return node->val;
    }
    
    void addAtHead(int val) {  // 时间复杂度为O(1)
        addAtIndex(0, val);  
    }
    
    void addAtTail(int val) {  // 时间复杂度为O(n)
        addAtIndex(size_, val);
    }
    
    void addAtIndex(int index, int val) {  // 时间复杂度为O(n)
        if (index > size_) {
            return;  
        }

        node* tmp = guard_.get();
        for (int i = 0; i < index; ++i) {
            tmp = tmp->next.get();
        }
        tmp->next = std::make_unique<node>(val, std::move(tmp->next));

        ++size_;  
    }
    
    void deleteAtIndex(int index) {  // 时间复杂度为O(n)
        if (!index_valid(index)) {
            return;
        }

        node* tmp = guard_.get();
        for (int i = 0; i < index; ++i) {
            tmp = tmp->next.get();
        }
        // 由于tmp->next为智能指针，当该指针无拥有者时内存将自动释放
        tmp->next = std::move(tmp->next->next);

        --size_;
    }

private:
    bool index_valid(int index) const 
    {
        return ((index >= 0) && (index < size_));
    }

private:
    std::unique_ptr<node> guard_;
    int size_;
};

时间复杂度与函数有关系，在代码中已进行注释。空间复杂度为O(1)。执行结果如下：