【算法-链表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,期待与你共同进步!

相关推荐
程序员Xu1 小时前
【LeetCode热题100道笔记】二叉树的右视图
笔记·算法·leetcode
笑脸惹桃花1 小时前
50系显卡训练深度学习YOLO等算法报错的解决方法
深度学习·算法·yolo·torch·cuda
阿维的博客日记2 小时前
LeetCode 48 - 旋转图像算法详解(全网最优雅的Java算法
算法·leetcode
GEO_YScsn2 小时前
Rust 的生命周期与借用检查:安全性深度保障的基石
网络·算法
程序员Xu2 小时前
【LeetCode热题100道笔记】二叉搜索树中第 K 小的元素
笔记·算法·leetcode
THMAIL3 小时前
机器学习从入门到精通 - 数据预处理实战秘籍:清洗、转换与特征工程入门
人工智能·python·算法·机器学习·数据挖掘·逻辑回归
Kevinhbr4 小时前
CSP-J/S IS COMING
数据结构·c++·算法
Armyyyyy丶4 小时前
Redis底层实现原理之五大基础结构
数据结构·redis·缓存
蕓晨4 小时前
set的插入和pair的用法
c++·算法
THMAIL4 小时前
深度学习从入门到精通 - AutoML与神经网络搜索(NAS):自动化模型设计未来
人工智能·python·深度学习·神经网络·算法·机器学习·逻辑回归