题解 | #LRU Cache#

LRU Cache

https://www.nowcoder.com/practice/3da4aeb1c76042f2bc70dbcb94513338

#include <iostream>
#include <map>
using namespace std;

class ListNode {
public:
    int val;
    int key;
    ListNode* pre;
    ListNode* next;
    ListNode(int ky, int va) :key(ky), val(va), pre(nullptr), next(nullptr) {}
    ListNode() :pre(nullptr), next(nullptr), key(-1), val(-1){}
};

class LRU {
private:
    map<int, int> kvMap;
    ListNode* head;
    ListNode* end;
    int size;
public:
    LRU() :head(nullptr),end(nullptr), size(0) {}
    LRU(int size) :head(nullptr), end(nullptr), size(size) {}
    int get(int key) {
        if (kvMap.find(key) == kvMap.end()) {
            return -1;
        }
        ListNode* p = head;
        ListNode* pre = nullptr;
        while (p) {
            if (p->key == key) {
                if (p == head && p->next) {
                    head = p->next;
                }
                else if ((p == head && !p->next )||(p == end)) {
                    return kvMap[key];
                }
                if (pre) {
                    pre->next = p->next;
                }
                if (p->next) {
                    p->next->pre = pre;
                }
                p->pre = end;
                end->next = p;
                p->next = nullptr;
                end = p;
            }
            pre = p;
            p = p->next;
        }
        return kvMap[key];
    }

    void put(int key, int val) {
        if (size == 0) {
            return;
        }
        if (kvMap.find(key) != kvMap.end()) {
            kvMap[key] = val;
            ListNode* p = head;
            while (p) {
                if (p->key == key) {
                    p->val = val;
                    return;
                }
                p = p->next;
            }
            return;
        }
        if (kvMap.size() + 1 > size) {
            ListNode* old = head;
            kvMap.erase(old->key);
            head = head->next;
            if (end == old) {
                end = nullptr;
            }
            delete old;
            
        }
        kvMap[key] = val;
        if (head == nullptr) {
            head = new ListNode(key, val);
        }
        else if (end == nullptr) {
            end = new ListNode(key, val);
            head->next = end;
            end->pre = head;
        }
        else {
            end->next = new ListNode(key, val);
            end->next->pre = end;
            end = end->next;
            
        }
    }

};

int main() {
    int size;
    cin >> size;
    LRU lru(size);
    char x;
    int k, v;
    while (cin >> x) {
        if (x == 'p') {
            cin >> k >> v;
            lru.put(k, v);
        } else {
            cin >> k;
            cout << lru.get(k) << endl;
        }
    }
}
// 64 位输出请用 printf("%lld")

写的不是很好,主要就是一个双向链表的更新,好像hash表也是不必要的

全部评论

相关推荐

练习生懒羊羊:开飞机把这个公司创飞吧
点赞 评论 收藏
分享
06-07 19:59
门头沟学院 C++
补药卡我啊😭:都快15年前的了还在11新特性
你的简历改到第几版了
点赞 评论 收藏
分享
屌丝逆袭咸鱼计划:心态摆好,man,晚点找早点找到最后都是为了提升自己好进正职,努力提升自己才是最关键的😤难道说现在找不到找的太晚了就炸了可以鸡鸡了吗😤早实习晚实习不都是为了以后多积累,大四学长有的秋招进的也不妨碍有的春招进,人生就这样
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-04 14:35
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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