代码随想录Day6-链表:力扣203、707题

203. 移除链表元素

题目链接
代码随想录文章讲解链接

方法一:直接移除

思路

直接移除满足条件的节点,但是对于头结点和非头结点的移除逻辑不同,所以要分别进行操作。

  • 时间复杂度: O ( n ) O(n) O(n)
  • 空间复杂度: O ( 1 ) O(1) O(1)
C++代码
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        // 删除头结点
        while (head != nullptr && head->val == val) {
            ListNode* tmp = head;
            head = head->next;
            delete tmp;
        }
        // 删除非头结点
        ListNode* cur = head;
        while (cur != nullptr && cur->next != nullptr) {
            if (cur->next->val == val) {
                ListNode* tmp = cur->next;
                cur->next = cur->next->next;
                delete tmp;
            } else {
                cur = cur->next;
            }
        }
        return head;
    }
};

方法二:使用虚拟头结点

思路

方法一需要分别处理头结点和非头结点,可以首先在链表头设置一个虚拟头结点,这样就能统一原本的头结点和非头结点的操作,使代码逻辑更加简洁,最后再返回虚拟头结点的next。

  • 时间复杂度: O ( n ) O(n) O(n)
  • 空间复杂度: O ( 1 ) O(1) O(1)
C++代码
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* preHead = new ListNode(0, head);
        ListNode* cur = preHead;
        while (cur->next != nullptr) {
            if (cur->next->val == val) {
                ListNode* tmp = cur->next;
                cur->next = cur->next->next;
                delete tmp;
            } else {
                cur = cur->next;
            }
        }
        head = preHead->next;
        delete preHead;
        return head;
    }
};

看完讲解的思考

代码实现遇到的问题

太久没处理各种数据结构了,对于指针的操作一开始没理清楚。


707. 设计链表

题目链接
代码随想录文章讲解链接

方法一:单链表

思路

经典单链表的操作,重点在于设置虚拟头结点,不设置虚拟头结点的话代码逻辑会很复杂。

  • 时间复杂度:get是 O ( i n d e x ) O(index) O(index),addAtHead是 O ( 1 ) O(1) O(1),addAtTail是 O ( n ) O(n) O(n),addAtIndex是 O ( i n d e x ) O(index) O(index),deleteAtIndex是 O ( i n d e x ) O(index) O(index)
  • 空间复杂度:不考虑链表本身的存储空间的话,所有操作都为 O ( 1 ) O(1) O(1)
C++代码
class MyLinkedList {
public:
    // 定义链表节点结构体
    struct LinkedNode {
        int val;
        LinkedNode* next;
        LinkedNode(int val) : val(val), next(nullptr) {}
    };

    MyLinkedList() {
        this->_size = 0;
        this->_dummyHead = new LinkedNode(-1);
    }
    
    int get(int index) {
        if (index < 0 || index >= this->_size) return -1;
        LinkedNode* cur = this->_dummyHead->next;
        for (int i = 0; i < index; ++i) cur = cur->next;
        return cur->val;
    }
    
    void addAtHead(int val) {
        LinkedNode* newNode = new LinkedNode(val);
        newNode->next = this->_dummyHead->next;
        this->_dummyHead->next = newNode;
        ++this->_size;
    }
    
    void addAtTail(int val) {
        LinkedNode* newNode = new LinkedNode(val);
        LinkedNode* cur = this->_dummyHead;
        while (cur->next != nullptr) cur = cur->next;
        cur->next = newNode;
        ++this->_size;
    }
    
    void addAtIndex(int index, int val) {
        if (index >= 0 && index <= this->_size) {
            LinkedNode* cur = this->_dummyHead;
            LinkedNode* newNode = new LinkedNode(val);
            for (int i = 0; i < index; ++i) cur = cur->next;
            newNode->next = cur->next;
            cur->next = newNode;
            ++this->_size;
        }
    }
    
    void deleteAtIndex(int index) {
        if (index >= 0 && index < this->_size) {
            LinkedNode* cur = this->_dummyHead;
            for (int i = 0; i < index; ++i) cur = cur->next;
            LinkedNode* delNode = cur->next;
            cur->next = delNode->next;
            delete delNode;
            delNode = nullptr;
            --this->_size;
        }
    }

private:
    int _size;
    LinkedNode* _dummyHead;
};

/**
 * 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);
 */

方法二:双向链表

思路

经典双向链表的操作,重点在于设置虚拟头结点和虚拟尾结点,不设置虚拟头尾结点的话代码逻辑会很复杂。

  • 时间复杂度:get是 O ( m i n { i n d e x , s i z e − 1 − i n d e x } ) O(min\{index, size-1-index\}) O(min{index,size1index}),addAtHead是 O ( 1 ) O(1) O(1),addAtTail是 O ( 1 ) O(1) O(1),addAtIndex是 O ( m i n { i n d e x , s i z e − i n d e x } ) O(min\{index, size-index\}) O(min{index,sizeindex}),deleteAtIndex是 O ( m i n { i n d e x , s i z e − i n d e x } ) O(min\{index, size-index\}) O(min{index,sizeindex})
  • 空间复杂度:不考虑链表本身的存储空间的话,所有操作都为 O ( 1 ) O(1) O(1)。但是双向链表相对于单链表需要额外的存储空间来存储prev
C++代码
class MyLinkedList {
public:
    // 定义双链表结点结构体
    struct DLinkedNode {
        int val;
        DLinkedNode* next;
        DLinkedNode* prev;
        DLinkedNode(int val) : val(val), next(nullptr), prev(nullptr) {}
        DLinkedNode(int val, DLinkedNode* next, DLinkedNode* prev) : val(val), next(next), prev(prev) {}
    };

    MyLinkedList() {
        this->_size = 0;
        this->head = new DLinkedNode(-1);
        this->tail = new DLinkedNode(-1);
        this->head->next = this->tail;
        this->tail->prev = this->head;
    }
    
    int get(int index) {
        if (index < this->_size && index >= 0) {
            DLinkedNode* cur;
            if (index < (this->_size - 1) / 2) {
                // 要获取的结点位于链表前半段
                cur = this->head->next;
                while (index--) cur = cur->next;
            } else {
                // 要获取的结点位于链表后半段
                cur = this->tail->prev;
                while (index++ < this->_size - 1) cur = cur->prev;
            }
            return cur->val;
        }
        return -1;
    }
    
    void addAtHead(int val) {
        DLinkedNode* newNode = new DLinkedNode(val, this->head->next, this->head);
        this->head->next->prev = newNode;
        this->head->next = newNode;
        ++this->_size;
    }
    
    void addAtTail(int val) {
        DLinkedNode* newNode = new DLinkedNode(val, this->tail, this->tail->prev);
        this->tail->prev->next = newNode;
        this->tail->prev = newNode;
        ++this->_size;
    }
    
    void addAtIndex(int index, int val) {
        if (index <= this->_size && index >= 0) {
            if (index <= this->_size / 2) {
                // 要增加的结点的位置位于链表前半段
                DLinkedNode* cur = this->head;
                while (index--) cur = cur->next;
                DLinkedNode* newNode = new DLinkedNode(val, cur->next, cur);
                cur->next->prev = newNode;
                cur->next = newNode;
            } else {
                // 要增加的结点的位置位于链表后半段
                DLinkedNode* cur = this->tail;
                while (index++ < this->_size) cur = cur->prev;
                DLinkedNode* newNode = new DLinkedNode(val, cur, cur->prev);
                cur->prev->next = newNode;
                cur->prev = newNode;
            }
            ++this->_size;
        }
    }
    
    void deleteAtIndex(int index) {
        if (index < this->_size && index >= 0) {
            DLinkedNode* delNode;
            if (index <= this->_size / 2) {
                // 要删除的结点位于链表前半段
                DLinkedNode* cur = this->head;
                while (index--) cur = cur->next;
                delNode = cur->next;
                delNode->next->prev = cur;
                cur->next = delNode->next;
            } else {
                // 要删除的结点位于链表后半段
                DLinkedNode* cur = this->tail;
                while (index++ < this->_size - 1) cur = cur->prev;
                delNode = cur->prev;
                delNode->prev->next = cur;
                cur->prev = delNode->prev;
            }
            delete delNode;
            delNode = nullptr;
            --this->_size;
        }
    }

private:
    int _size;
    DLinkedNode* head;
    DLinkedNode* tail;
};

/**
 * 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);
 */

看完讲解的思考

划重点:单链表要设置虚拟头结点!双向链表要设置虚拟头尾结点
可以大大简化代码逻辑。
对某个结点进行删除或者在某个结点之后添加新结点的操作,一定是遍历到目标结点的前一个结点进行操作。例如:要删除第5个结点,则要遍历到第4个结点,然后对第5个结点进行删除;要在第5个位置添加结点,则要遍历到第4个结点,然后在后面添加新结点。
该题还可以用其他链表实现,例如循环链表,但目前水平有限,之后二刷再来填坑。

代码实现遇到的问题

因为没有设置虚拟头结点,导致代码逻辑混乱,一堆bug…


最后的碎碎念

今日的数据结构链表太久没写了,折腾了半天,第二题搞了一晚上,今天就刷两题吧。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值