LRU 缓存机制c++ 哈希表+双向链表

设计LRU缓存结构

http://www.nowcoder.com/questionTerminal/e3769a5f49894d49b871c09cadd13a61

#include <unordered_map>
struct DListNode{
    int key;
    int val;
    DListNode* prev;//前驱
    DListNode* next;//后继
    DListNode(): key(0), val(0), prev(nullptr), next(nullptr) {}
    DListNode(int _key,int _val): key(_key), val(_val), prev(nullptr), next(nullptr) {}
};

class Solution {
private:
    unordered_map<int, DListNode*> cache;
    DListNode* head;
    DListNode* tail;
    //最大容量
    int capacity;
public:
    Solution(){
        // 使用伪头部和伪尾部节点
        head = new DListNode();
        tail = new  DListNode();
        head->next = tail;
        tail->prev = head;
        capacity = 0;
    }
    vector<int> LRU(vector<vector<int> >& operators, int k) {
        // write code here
        vector<int> res;
        capacity = k;
        for(int i = 0;i<operators.size();i++){
            if(operators[i].size()==3){//set(x, y)
                set(operators[i][1],operators[i][2]);
            }else{
                res.push_back(get(operators[i][1]));
            }
        }
        return res;
    }
    int get(int key) {
        //查找不存在 返回-1
        if (!cache.count(key)) {
            return -1;
        }
        // 如果 key 存在,先通过哈希表定位,再移到头部
        DListNode* node = cache[key];
        moveToHead(node);
        return node->val;
    }

    void set(int key, int value) {
        if (!cache.count(key)) {
            // 如果 key 不存在,创建一个新的节点
            DListNode* node = new DListNode(key, value);
            // 添加进哈希表
            cache[key] = node;
            // 添加至双向链表的头部
            addToHead(node);
            if (cache.size() > capacity) {
                // 如果超出容量,删除双向链表的尾部节点
                DListNode* removed = removeTail();
                // 删除哈希表中对应的项
                cache.erase(removed->key);
                // 防止内存泄漏
                delete removed;
            }
        }
        else {
            // 如果 key 存在,先通过哈希表定位,再修改 value,并移到头部
            DListNode* node = cache[key];
            node->val = value;
            moveToHead(node);
        }
    }

    void addToHead(DListNode* node) {
        //添加到链表头部
        node->prev = head;
        node->next = head->next;
        head->next->prev = node;
        head->next = node;
    }

    void removeNode(DListNode* node) {
        //删除
        //前驱节点的后驱 = node的后驱
        //后驱节点的前驱 = node的前驱
        node->prev->next = node->next;
        node->next->prev = node->prev;
    }

    void moveToHead(DListNode* node) {
        //移动到头部
        removeNode(node);
        addToHead(node);
    }

    //删除尾端节点,并返回
    DListNode* removeTail() {
        DListNode* node = tail->prev;
        removeNode(node);
        return node;
    }
};
全部评论

相关推荐

07-17 12:09
门头沟学院 Java
讲的口干舌燥,头都晕了怎么要讲这么长啊
码农索隆:没事,你口干舌燥,他不一定会看,
投递小鹏汽车等公司7个岗位
点赞 评论 收藏
分享
06-26 22:20
门头沟学院 Java
码农索隆:让你把简历发给她,她说一些套话,然后让你加一个人,说这个人给你改简历,然后开始卖课
我的求职精神状态
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-21 13:38
8月实习会变多吗现在还没找到实习该怎么办...回复的hr好少
码农索隆:3-4月就要开始找,基本上6月份就发offer,7月初已经开始暑期实习了。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务