146.LRU缓存-图解LRU

LRU缓存是一种满足最近最少使用约束的数据结构。我们可以用一个简单的例子来理解:假设你有一摞书,最多只能放capacity本。当你需要找一本书时,如果书在摞中,就返回它的版本(即key-value);如果不在,就返回-1。当你想放入一本新书时,如果这本书已经存在,就更新它的版本号;如果不存在,就把新书放在最上面。如果书的数量超过了capacity,就把最下面那本书移出。

那么,在这个例子中,我们主要用到了哪些操作呢?又该用什么数据结构来实现呢?由于题目要求get()和put()的时间复杂度为O(1),并且需要同时存放key-value,还要删除最久未使用的元素,因此可以使用双向链表来解决。具体来说,我们主要用到了以下操作:

1.删除

将一个节点删除

2.将节点放在最前面

3.快速找出要找的节点

使用哈希表,用key与节点作映射

cpp 复制代码
class Node {
public:
    int key;
    int value;
    Node *prev;
    Node *next;
    Node(int k = 0,int v = 0):key(k),value(v){};
};

class LRUCache {
public:
    int capacity = 0;
    Node *dummy;
    unordered_map<int,Node*> key_to_node;
    //删除节点
    void remove(Node *x) {
        x->prev->next = x->next;
        x->next->prev = x->prev;
    }
    //将节点放在最前
    void push_front(Node *x) {
        x->prev = dummy;
        x->next = dummy->next;
        x->next->prev = x;
        x->prev->next = x;
    }
    //获取节点
    Node* getNode(int key) {
        auto it = key_to_node.find(key);
        if (it == key_to_node.end()) {
            return nullptr;
        }
        Node *node = key_to_node[key];
        remove(node);
        push_front(node);
        return node;
    }
  
    LRUCache(int capacity) : capacity(capacity),dummy(new Node()) {
        dummy->next = dummy;
        dummy->prev = dummy;
    }
    
    int get(int key) {
        Node *node = getNode(key);
        return node ? node->value : -1;
    }
    
    void put(int key, int value) {
        Node *node = getNode(key);
        if (node) {
            node->value = value;
            return;
        }
        key_to_node[key] = node = new Node(key,value);
        push_front(node);
        if (capacity < key_to_node.size()) {
            //最久未使用节点
            Node *back_node = dummy->prev;
            key_to_node.erase(back_node->key);
            remove(back_node);
            //释放内存
            delete back_node;
        }
    
    }
};

时间复杂度:O(1)

空间复杂度:O(min(p,capacity),p为put的次数

相关推荐
燃于AC之乐1 小时前
我的算法修炼之路--4 ———我和算法的爱恨情仇
算法·前缀和·贪心算法·背包问题·洛谷
MM_MS7 小时前
Halcon变量控制类型、数据类型转换、字符串格式化、元组操作
开发语言·人工智能·深度学习·算法·目标检测·计算机视觉·视觉检测
独自破碎E7 小时前
【二分法】寻找峰值
算法
mit6.8247 小时前
位运算|拆分贪心
算法
ghie90908 小时前
基于MATLAB的TLBO算法优化实现与改进
开发语言·算法·matlab
恋爱绝缘体18 小时前
2020重学C++重构你的C++知识体系
java·开发语言·c++·算法·junit
wuk9988 小时前
VSC优化算法MATLAB实现
开发语言·算法·matlab
Z1Jxxx8 小时前
加密算法加密算法
开发语言·c++·算法
乌萨奇也要立志学C++8 小时前
【洛谷】递归初阶 三道经典递归算法题(汉诺塔 / 占卜 DIY/FBI 树)详解
数据结构·c++·算法
vyuvyucd9 小时前
C++引用:高效编程的别名利器
算法