Leetcode146. LRU 缓存 -hot100

题目:


代码(首刷看解析 2024年3月5日):

        学的灵神,太优美了,这个哨兵节点思想

class Node {
public:
    int key,value;
    Node* prev,*next;

    Node(int k = 0, int v = 0) : key(k), value(v) {}
};

class LRUCache {
private:
    int capacity;
    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->next = dummy->next;
        x->prev = dummy;
        x->prev->next = x;
        x->next->prev = x;
    }
    Node* get_node(int key) {
        auto it = key_to_node.find(key); 
        if (it == key_to_node.end())
            return nullptr;
        auto node = it->second;
        remove(node);
        push_front(node);
        return node;
    }
public:
    LRUCache(int capacity) : capacity(capacity) , dummy(new Node){
        dummy->prev = dummy;
        dummy->next = dummy;
    }
    
    int get(int key) {
        auto node = get_node(key);
        return node ? node->value : -1;
    }
    
    void put(int key, int value) {
        auto node = get_node(key);
        if (node) {
            node->value = value;
            return;
        }
        node = new Node(key, value);
        key_to_node[key] = node;
        push_front(node);
        if (key_to_node.size() > capacity) {
            auto back_node = dummy->prev;
            key_to_node.erase(back_node->key);
            remove(back_node);
            delete back_node;
        }
    }
};

/**
 * Your LRUCache object will be instantiated and called as such:
 * LRUCache* obj = new LRUCache(capacity);
 * int param_1 = obj->get(key);
 * obj->put(key,value);
 */

代码(二刷看解析 2024年4月14日)

struct Node{
    int key;
    int value;
    Node* prev;
    Node* next;
    Node(int k = 0, int v = 0) : key(k), value(v){}
};

class LRUCache {
protected:
    Node* dummyHead;
    int _capacity;
    unordered_map<int,Node*> key_to_node;
private:
    void remove(Node* cur) {
        cur->prev->next = cur->next;
        cur->next->prev = cur->prev;
    }
    void push_front(Node* cur) {
        cur->next = dummyHead->next;
        cur->prev = dummyHead;
        cur->next->prev = cur;
        cur->prev->next = cur;
    }
    Node* get_node(int key) {
        auto it = key_to_node.find(key);
        if (it == key_to_node.end()) {
            return nullptr;
        }
        remove(it->second);
        push_front(it->second);
        return it->second;
    }
    
public:
    LRUCache(int capacity):_capacity(capacity), dummyHead(new Node) {
        dummyHead->next = dummyHead;
        dummyHead->prev = dummyHead;
    }
    
    int get(int key) {
        auto node = get_node(key);
        return node == nullptr ? -1 : node->value;
    }
    
    void put(int key, int value) {
        // 不存在,则创建
        auto node = get_node(key);
        if (node) {
            node->value = value;
            return;
        }
        node = new Node(key, value);
        key_to_node[key] = node;
        push_front(node);
        // 若超过大小了,删除最后面的
        if (key_to_node.size() > _capacity) {
            Node* tmp = dummyHead->prev;
            key_to_node.erase(tmp->key);
            remove(tmp);
            delete tmp;   
        }
    }
};

  • 11
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值