用Python实现LRU Cache(链表+哈希表)

class Node:
    def __init__(self, key, val):
        self.before = None
        self.after = None
        self.key = key
        self.val = val

class LRUCache:

    def __init__(self, capacity: int):
        self.k = capacity
        self.cache = {}
        self.head = None
        self.tail = None

    def get(self, key: int) -> int:
        if key not in self.cache:
            return -1
        node = self.cache[key]
        if self.tail == node and self.tail.before is not None:
            self.tail = self.tail.before
        if self.head is not node:
            pre = node.before
            nex = node.after
            if pre:
                pre.after = node.after
            if nex:
                nex.before = node.before
            self.head.before = node
            node.after = self.head
            node.before = None
            self.head = node
        return node.val


    def put(self, key: int, value: int) -> None:
        n = Node(key, value)
        if key in self.cache:
            self.cache[key].val = value
            self.get(key)
        else:
            if len(self.cache) == self.k:
                self.cache.pop(self.tail.key)
                self.tail = self.tail.before
                if self.tail:
                    self.tail.after = None
            if self.tail is None:
                self.tail = n

            n.after = self.head
            if self.head:
                self.head.before = n
            self.head = n
            self.cache[n.key] = n




  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值