【算法-链表1】移除链表元素 和 设计链表

今天,带来链表相关算法的讲解。文中不足错漏之处望请斧正!

理论基础点这里


移除链表元素

1. 思路

链表的删除和插入,都是改变链接关系。

如图:要删除值为2的结点,要找到其上一个节点,让上一个节点指向值为3的结点。

但这是一个单链表,我们怎么找到上一个呢?只好用一个cur指针从头开始走,cur→next就是可能要删除的节点。

1.1 在原链表删除

我们还需要考虑边界情况:删除头结点。此时只要让head向后走。

但是,这样我们的删除方法就不统一了,头结点是一种删法,其他结点又是另一种删法,容易让代码变多,逻辑也不够简单。

1.2 虚拟头结点

想要统一其实很简单,我们最重要的就是找上一个,但是要删除头结点的时候找不到上一个,那就再给一个虚拟头结点,让它指向真正头结点。头结点就有"上一个结点"了。

此时,dummyhead→next其实就是真正的head,删除方法也能统一。

2. 参考代码

原链表删除

cpp 复制代码
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode *dying = nullptr;
        // 移除头部需移除的元素
        while (head != nullptr && head->val == val) {
            dying = head;
            head = head->next;
            delete dying;
        }

        // cur不可能是需要移除的元素,需要移除的元素只可能是cur->next
        ListNode *cur = head; 
        while (cur != nullptr && cur->next != nullptr) {
            if (cur->next->val == val) {
                dying = cur->next;
                cur->next = cur->next->next;
                delete dying;
            } else {
                cur = cur->next;
            }
        }

        return head;
    }
};

需要注意:当我们删除某个节点时,cur不需要变,下一次要探测的结点恰好就是cur→next→next。

虚拟头节点

cpp 复制代码
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) 
    {
        ListNode* dummy = new ListNode(-1, head);
        ListNode* cur = dummy;
        while(cur->next)
        {
            if(cur->next->val == val)
            {
                ListNode* tmp = cur->next;
                cur->next = cur->next->next;
                delete tmp;
            }
            else cur = cur->next;
        }
        ListNode* result = dummy->next;
        delete dummy;
        return result;
    }
};

设计链表

1. 思路

链表操作我们都了解了,就是修改某些节点之间的链接关系,按题目要求做即可。这里实现带虚拟头节点的链表。

2. 参考代码

cpp 复制代码
class MyLinkedList {
public:
    MyLinkedList() {
        _dummy = new ListNode(-1);
        _size = 0;
    }

    void print() {
        ListNode *cur = _dummy->__next;
        while (cur) {
            cout << cur->__val << " ";
            cur = cur->__next;
        }
        cout << endl;
    }
    
    int get(int index) {
        if (index >= _size) return -1;

        ListNode *cur = _dummy->__next;
        for (int i = 0; i < index; ++i) cur = cur->__next;
        return cur->__val;
    }
    
    void addAtHead(int val) {
        ListNode *newNode = new ListNode(val, _dummy->__next);
        _dummy->__next = newNode;
        ++_size;
    }
    
    void addAtTail(int val) {
        ListNode *tail = _dummy;
        while (tail->__next != nullptr) tail = tail->__next;

        ListNode *newNode = new ListNode(val);
        tail->__next = newNode;
        ++_size;
    }
    
    // 将一个值为 val 的节点插入到链表中下标为 index 的节点之前
    void addAtIndex(int index, int val) {
        if (index > _size) return;

        ListNode *prev = _dummy;
        for (int i = 0; i < index; ++i) prev = prev->__next;

        ListNode *newNode = new ListNode(val, prev->__next);
        prev->__next = newNode;
        ++_size;
    }
    
    void deleteAtIndex(int index) {
        if (index >= _size) return;

        ListNode *prev = _dummy;
        ListNode *cur = prev->__next;
        for (int i = 0; i < index; ++i) {
            prev = cur;
            cur = cur->__next;
        }
        ListNode *dying = cur;
        prev->__next = cur->__next;
        delete dying;
        --_size;
    }
private:
    struct ListNode {
        int __val;
        ListNode *__next;
        ListNode(int val, ListNode *next = nullptr) 
        : __val(val), __next(next) {}
    };
private:
    ListNode *_dummy;
    size_t _size;
};

今天的分享就到这里了,感谢您能看到这里。

这里是培根的blog,期待与你共同进步!

相关推荐
聚客AI13 小时前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v15 小时前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工17 小时前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农18 小时前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了19 小时前
AcWing学习——双指针算法
c++·算法
moonlifesudo19 小时前
322:零钱兑换(三种方法)
算法
NAGNIP2 天前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队2 天前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
Fanxt_Ja2 天前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下2 天前
最终的信号类
开发语言·c++·算法