leetcode 707. 设计链表

leetcode 707. 设计链表

题目描述：

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

解题分析：

单链表和双链表在时间复杂度上一样，单链表在空间上更好，用单链表来实现。
思路来自y总，首先构建了链表节点结构体，并对其初始化。
1.获取链表第index个节点的数值，如果 index 小于0或者头节点无效，就返回 -1；否则顺序遍历到索引为index的结点，返回val。
2.在链表的最前面插入一个节点，直接定义当前节点，并将它与头节点相连。
3.在链表的最后面插入一个节点，如果链表为空，链表头指向当前节点；否则顺序遍历到最后结点。
4.在链表第index个节点前面插入一个节点，如果index<=0，则相当于头插；如果index与链表长度相当，相当于尾插；否则顺序遍历到index - 1，断开index - 1与index，连接新结点和index，再连接index - 1与新结点。
5.删除链表的第index个节点，索引index有效，即index>=0且index<len，然后顺序遍历到index-1，将index - 1直接与index + 1相连接。

class MyLinkedList {
public:
    //定义链表
    struct Node {
        int val;
        Node* next;//指针
        Node(int val): val(val), next(NULL) {}//构造函数
    }*head;

    MyLinkedList() {
        head = NULL;//定义头节点
    }
    
    int get(int index) {
        if (index < 0) return -1;
        auto p = head;
        for (int i = 0; i < index && p; i++) p = p->next;
        if (!p) return -1; //p无解
        return p->val;
    }
    
    void addAtHead(int val) {
        //头插
        auto cur = new Node(val);//定义当前节点
        cur->next = head;
        head = cur;
    }
    
    void addAtTail(int val) {
        //尾插
        if (!head) head = new Node(val); //链表为空，链表头指向当前节点
        else {
            auto p = head;
            while (p->next) p = p->next; //最后p为尾节点
            p->next = new Node(val);
        }
    }

    void addAtIndex(int index, int val) {
        if (index <= 0) addAtHead(val);
        else {
            int len = 0;
            for (auto p = head; p; p = p->next) len++;
            if (index == len) addAtTail(val);
            else if (index < len){
                auto p = head;
                for (int i = 0; i < index - 1; i++) p = p->next; // 找到index的前一个下标
                auto cur = new Node(val);
                cur->next = p->next;
                p->next = cur;
            }
        }
    }
    
    void deleteAtIndex(int index) {
        int len = 0;
        for (auto p = head; p; p = p->next) len++;
        if (index >= 0 && index < len) {
            if (!index) head = head->next;
            else {
                auto p = head;
                for (int i = 0; i < index - 1; i++) p = p->next;
                p->next = p->next->next;
            }
        }
    }
};

/**
 * Your MyLinkedList object will be instantiated and called as such:
 * MyLinkedList* obj = new MyLinkedList();
 * int param_1 = obj->get(index);
 * obj->addAtHead(val);
 * obj->addAtTail(val);
 * obj->addAtIndex(index,val);
 * obj->deleteAtIndex(index);
 */