题目
LRU 缓存机制,设计和实现一个 LRU(最近最少使用)缓存数据结构。
LRUCache(int capacity) 以正整数作为容量 capacity 初始化 LRU 缓存。
get(key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回一个默认值。
put(key, value) 如果关键字已经存在,则变更其数据值;如果关键字不存在,则插入该组「key-value」。当缓存容量达到上限时,它应该在写入新数据之前删除最久未使用的数据值,从而为新的数据值留出空间
时间复杂度要求O(1)
思考
First
要实现在O(1)的时间复杂度里获取key
对应的value
,那么数据结构就应该是使用HashMap
,也就是Python中的Dict
Second
还要知道最久未使用的数据,可以考虑用链表,但是单链表在插入到链表尾部的时间复杂度为O(n),所以不考虑,应当使用双向链表。
Third
使用Dict
来定位节点在链表中的位置,记录下来双向链表的头节点与尾节点,无论是插入还是删除一个节点,时间复杂度都为O(1)
实现
双向链表
使用__slots__
可以用来节约内存。
LRUCache
get
操作先获取key
在双向链表中对应的节点,如果有,就将该节点从头部删除,并插入尾部。否则返回默认值
put
操作先判断key是否存在,若不存在,就新建一个节点,并判断空间是否足够,如果不足够就删去尾部的一个节点;
若存在,就更新该节点的值,并将该节点移动至头部
class Node(object):
__slots__ = ["key", "value", "prev", "next"]
def __init__(self, key, value):
self.key = key
self.value = value
self.prev = None