实现LRU缓存机制

题目

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
        self.next = None


class LRUCache(object):

    def __init__(self, capacity=0):

        self.dic = {}
        self.capacity = capacity
        self.head = Node(-1, -1)
        self.tail = Node(-1, -1)

        self.head.next = self.tail
        self.tail.prev = self.head
        self.size = 0

    def get(self, key, default=None):
        if key not in self.dic:
            return default

        node = self.dic[key]
        self.move_to_head(node)
        return node.value

    def put(self, key, value):
        if key not in self.dic:
            node = Node(key=key, value=value)
            self.dic[key] = node
            self.add_to_head(node)
            self.size += 1
            if self.size > self.capacity:
                removed = self.remove_tail()
                self.dic.pop(removed.key)
                self.size -= 1
        else:
            node = self.dic[key]
            node.value = value
            self.move_to_head(node)

    def remove(self, node):
        node.prev.next = node.next
        node.next.prev = node.prev

    def remove_tail(self):
        node = self.tail.prev
        self.remove(node)
        return node

    def move_to_head(self, node):
        self.remove(node)
        self.add_to_head(node)

    def add_to_head(self, node):
        node.prev = self.head
        node.next = self.head.next
        self.head.next.prev = node
        self.head.next = node


if __name__ == "__main__":
    dic = LRUCache(3)
    dic.put('a', 1)
    dic.put('b', 2)
    dic.put('c', 3)
    dic.put('d', 4)

    print(dic.get('a'))