146. LRU 缓存机制(双向链表+哈希表)

难度:中等

运用你所掌握的数据结构,设计和实现一个 LRU (最近最少使用) 缓存机制 。
实现 LRUCache 类:
LRUCache(int capacity) 以正整数作为容量 capacity 初始化 LRU 缓存
int get(int key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回 -1 。
void put(int key, int value) 如果关键字已经存在,则变更其数据值;如果关键字不存在,则插入该组「关键字-值」。当缓存容量达到上限时,它应该在写入新数据之前删除最久未使用的数据值,从而为新的数据值留出空间。
进阶:你是否可以在 O(1) 时间复杂度内完成这两种操作?

示例:

输入
[“LRUCache”, “put”, “put”, “get”, “put”, “get”, “put”, “get”, “get”, “get”]
[[2], [1, 1], [2, 2], [1], [3, 3], [2], [4, 4], [1], [3], [4]]
输出
[null, null, null, 1, null, -1, null, -1, 3, 4]
解释
LRUCache lRUCache = new LRUCache(2);
lRUCache.put(1, 1); // 缓存是 {1=1}
lRUCache.put(2, 2); // 缓存是 {1=1, 2=2}
lRUCache.get(1); // 返回 1
lRUCache.put(3, 3); // 该操作会使得关键字 2 作废,缓存是 {1=1, 3=3}
lRUCache.get(2); // 返回 -1 (未找到)
lRUCache.put(4, 4); // 该操作会使得关键字 1 作废,缓存是 {4=4, 3=3}
lRUCache.get(1); // 返回 -1 (未找到)
lRUCache.get(3); // 返回 3
lRUCache.get(4); // 返回 4

解题思路:
本题使用双向链表和哈希表来解决。使用链表是为了表示头部的键值对是最近使用的,而尾部的键值对是最久没有使用的。那为什么不使用单链表呢?因为一旦删除了尾结点,我们无法知道尾结点的上一个节点是谁,所以需要在尾结点前加入pre指向前一个节点,也就是双向链表。使用哈希表的目的是可以在O(1)时间内找到key所对应的节点,这个节点存有key所对应的value。

python代码:

# 双向链表+哈希表
class DLinkedNode:
    def __init__(self, key=0, value=0) -> None:
        self.key = key
        self.value = value
        self.prev = None
        self.next = None

class LRUCache:
    def __init__(self, capacity: int):
        self.capacity = capacity
        self.hash_map = dict()
        self.head = DLinkedNode()
        self.tail = DLinkedNode()
        self.head.next = self.tail
        self.tail.prev = self.head
        self.size = 0

    def get(self, key: int) -> int:
        if key in self.hash_map:
            self.removeNode(self.hash_map[key])
            self.moveTohead(self.hash_map[key])
            return self.hash_map[key].value
        else:
            return -1

    def put(self, key: int, value: int) -> None:
        if key in self.hash_map:
            self.hash_map[key].value = value
            self.removeNode(self.hash_map[key])
            self.moveTohead(self.hash_map[key])
        else:
            Node = DLinkedNode(key, value)
            if self.size < self.capacity:
                self.size += 1
            else:
                self.hash_map.pop(self.tail.prev.key)
                self.removeNode(self.tail.prev)
            self.hash_map[key] = Node
            self.moveTohead(Node)

    def removeNode(self, Node):
        Node.next.prev = Node.prev
        Node.prev.next = Node.next

    def moveTohead(self, Node):
        Node.next = self.head.next
        self.head.next.prev = Node
        self.head.next = Node
        Node.prev = self.head



# Your LRUCache object will be instantiated and called as such:
# obj = LRUCache(capacity)
# param_1 = obj.get(key)
# obj.put(key,value)

复杂度分析:

  • 时间复杂度 O ( 1 ) O(1) O(1):查找、添加、删除、移动到头节点(插入),都是常数时间;
  • 空间复杂度 O ( c a p a c i t y ) O(capacity) O(capacity):双向链表和哈希表最多存储 c a p a c i t y capacity capacity个节点。
上一篇:Linux安装python模块


下一篇:【LeetCode】146. LRU 缓存机制