146. LRU 缓存 && 带TTL的LRU缓存实现(拓展)

LRU缓存

方法一:手动实现双向链表 + 哈希表

cpp 复制代码
struct Node{
    int val;
    int key;
    Node* prev;
    Node* next;
    Node(int a, int b): key(a), val(b), prev(nullptr), next(nullptr) {}
    Node():key(0), val(0), prev(nullptr), next(nullptr) {}
};
class LRUCache {
private:
    Node* removeTail() {
        Node* node = tail->prev;
        removeNode(node);
        return node;
    }

    void removeNode(Node* node) {
        node->next->prev = node->prev;
        node->prev->next = node->next;
    }

    void addToHead(Node* node) {
        node->next = head->next;
        node->prev = head;
        head->next = node;
        node->next->prev = node;
    }
public:
    unordered_map<int, Node*> cache;
    int size;
    int capacity;
    Node* head;
    Node* tail;
    LRUCache(int capacity) {
        this->capacity = capacity;
        size = 0;
        head = new Node();
        tail = new Node();
        head->next = tail;
        tail->prev = head;
    }
    
    int get(int key) {
        if(!cache.count(key)) {
            return -1;
        }
        Node* node = cache[key];
        removeNode(node);
        addToHead(node);
        return node->val;
    }
    
    void put(int key, int value) {
        if(!cache.count(key)) {
            Node* node = new Node(key, value);
            addToHead(node);
            ++size;
            cache[key] = node;
            if(size > capacity) {
                Node* removed = removeTail();
                cache.erase(removed->key);
                --size;
                delete removed;
            }
        } else {
            Node* node = cache[key];
            node->val = value;
            removeNode(node);
            addToHead(node);
        }
    }
};

拓展

实现带TTL的LRU缓存
要求 :

在本题的基础上,为 key 增加过期时间(put 调用时额外传入过期时间)。如果 key 过期,则需要删除掉。

cpp 复制代码
#include <unordered_map>
#include <chrono>

using namespace std;
using namespace std::chrono;

struct Node {
    int val;
    int key;
    Node* prev;
    Node* next;
    time_point<steady_clock> timestamp; // 时间戳,记录节点插入或更新时间

    Node(int a, int b) : key(a), val(b), prev(nullptr), next(nullptr) {
        timestamp = steady_clock::now(); // 初始化时间戳
    }

    Node() : key(0), val(0), prev(nullptr), next(nullptr) {
        timestamp = steady_clock::now(); // 初始化时间戳
    }
};

class LRUCache {
private:
    Node* removeTail() {
        Node* node = tail->prev;
        removeNode(node);
        return node;
    }

    void removeNode(Node* node) {
        node->next->prev = node->prev;
        node->prev->next = node->next;
    }

    void addToHead(Node* node) {
        node->next = head->next;
        node->prev = head;
        head->next = node;
        node->next->prev = node;
    }

    bool isExpired(Node* node, int ttl) {
        auto now = steady_clock::now();
        auto duration = duration_cast<milliseconds>(now - node->timestamp).count();
        return duration > ttl; // 如果超过TTL,则认为已过期
    }

public:
    unordered_map<int, Node*> cache;
    int size;
    int capacity;
    int ttl; // TTL时间,单位为毫秒
    Node* head;
    Node* tail;

    LRUCache(int capacity, int ttl) : capacity(capacity), ttl(ttl) {
        size = 0;
        head = new Node();
        tail = new Node();
        head->next = tail;
        tail->prev = head;
    }

    int get(int key) {
        if (!cache.count(key)) {
            return -1;
        }
        Node* node = cache[key];

        // 检查是否过期
        if (isExpired(node, ttl)) {
            removeNode(node);
            cache.erase(key);
            --size;
            delete node;
            return -1;
        }

        // 更新节点时间戳
        node->timestamp = steady_clock::now();
        removeNode(node);
        addToHead(node);
        return node->val;
    }

    void put(int key, int value) {
        if (!cache.count(key)) {
            Node* node = new Node(key, value);

            // 如果缓存已满,移除最久未使用的节点
            if (size >= capacity) {
                Node* removed = removeTail();
                cache.erase(removed->key);
                --size;
                delete removed;
            }

            // 插入新节点
            addToHead(node);
            cache[key] = node;
            ++size;
        } else {
            Node* node = cache[key];
            node->val = value;

            // 检查是否过期
            if (isExpired(node, ttl)) {
                removeNode(node);
                cache.erase(key);
                --size;
                delete node;
                put(key, value); // 重新插入
                return;
            }

            // 更新节点时间戳
            node->timestamp = steady_clock::now();
            removeNode(node);
            addToHead(node);
        }
    }
};

总结 :

在原代码基础上为每一个steady_lock的timestamp时间戳, 每次get或者put已有Node时进行检测isExpired,如果检测已经过期则对get删除节点, 对put更新时间戳

相关推荐
GEEK零零七13 分钟前
Leetcode 1103. 分糖果 II
数学·算法·leetcode·等差数列
今天背单词了吗98029 分钟前
算法学习笔记:7.Dijkstra 算法——从原理到实战,涵盖 LeetCode 与考研 408 例题
java·开发语言·数据结构·笔记·算法
苦夏木禾2 小时前
js请求避免缓存的三种方式
开发语言·javascript·缓存
重庆小透明2 小时前
力扣刷题记录【1】146.LRU缓存
java·后端·学习·算法·leetcode·缓存
desssq2 小时前
力扣:70. 爬楼梯
算法·leetcode·职场和发展
clock的时钟3 小时前
暑期数据结构第一天
数据结构·算法
小小小小王王王3 小时前
求猪肉价格最大值
数据结构·c++·算法
岁忧3 小时前
(LeetCode 面试经典 150 题 ) 58. 最后一个单词的长度 (字符串)
java·c++·算法·leetcode·面试·go
Java初学者小白3 小时前
秋招Day14 - Redis - 应用
java·数据库·redis·缓存
BIYing_Aurora3 小时前
【IPMV】图像处理与机器视觉:Lec13 Robust Estimation with RANSAC
图像处理·人工智能·算法·计算机视觉