其他分享
首页 > 其他分享> > LRUCache

LRUCache

作者:互联网

# LRU: Least recently used,最近最少使用

 

【一般使用链表+map这两个数据结构来实现】

# 使用map来加快缓存数据的查找

# 使用链表来记录最近的访问情况,最近被访问了(Get和Put都属于是最近访问了),就会被移到链表头

# 当Put进去的数据超过缓存容量时,链表尾部的元素就会被移出缓存

 

local LRUCache = {}
LRUCache.__index = LRUCache

function LRUCache.new(capacity)
    local obj = {}
    setmetatable(obj, LRUCache)
    obj:ctor(capacity)
    return obj
end

function LRUCache:ctor(capacity)
    self.capacity = capacity
    self.cache = {}
    self.list = LinkedList.new()
end

function LRUCache:Put(key, value)
    local node = self.cache[key]
    if nil == node then --缓存中没有
        node = self.list:AddFirst({k=key, v=value})
        self.cache[key] = node
        --如果超过了容量, 移除链表尾部元素
        if self.list:GetCount() > self.capacity then
            local nodeValue = self.list:RemoveLast():GetValue()
            self.cache[nodeValue.k] = nil
            --print(">capacity", nodeValue.k, nodeValue.v)
        end
    else --缓存中已经有了
        node:GetValue().v = value
        --最近被访问了, 放到链表最前面
        self.list:MoveToFirst(node)
    end
end

function LRUCache:Get(key)
    local node = self.cache[key]
    if nil == node then return nil end --缓存中没有
    self.list:MoveToFirst(node)
    return node:GetValue().v
end

function LRUCache:Remove(key)
    local node = self.cache[key]
    if nil == node then return end --缓存中没有
    self.cache[key] = nil
    self.list:MoveToFirst(node)
end

function LRUCache:Clear()
    self.cache = {}
    self.list:Clear()
end

 

【链表部分】

local Node = {}
Node.__index = Node

function Node.new(list, value)
    local obj = {}
    setmetatable(obj, Node)
    obj:ctor(list, value)
    return obj
end

function Node:ctor(list, value)
    self.list = list
    self.value = value
    self.next = nil
    self.prev = nil
end

function Node:SetValue(value)
    self.value = value
end
function Node:GetValue()
    return self.value
end

function Node:GetNext()
    return self.next
end
function Node:GetPrev()
    return self.prev
end

 

local LinkedList = {}
LinkedList.__index = LinkedList

function LinkedList.new()
    local obj = {}
    setmetatable(obj, LinkedList)
    obj:ctor()
    return obj
end

function LinkedList:ctor()
    self.head = nil
    self.tail = nil
    self.count = 0
end

function LinkedList:GetCount()
    return self.count
end

function LinkedList:Clear()
    if 0 == self.count then return end

    local allNodes = {}
    local node = self.head
    while nil ~= node do
        node.list = nil
        node.prev = nil
        local tempNext = node.next
        node.next = nil
        table.insert(allNodes, node)
        node = tempNext
    end
    self.head = nil
    self.tail = nil
    self.count = 0
    return allNodes --所有游离状nodes
end

function LinkedList:GetFirst()
    if 0 == self.count then return nil end
    return self.head
end
function LinkedList:GetFirstValue()
    if 0 == self.count then return nil end
    return self.head:GetValue()
end

function LinkedList:GetLast()
    if 0 == self.count then return nil end
    return self.tail
end
function LinkedList:GetLastValue()
    if 0 == self.count then return nil end
    return self.tail:GetValue()
end

function LinkedList:AddFirst(value)
    if 0 == self.count then
        self.count = 1
        self.head = Node.new(self, value)
        self.tail = self.head
        return self.head
    end
    self.count = self.count + 1
    local newHead = Node.new(self, value)
    newHead.next = self.head
    self.head.prev = newHead
    self.head = newHead
    return newHead
end

function LinkedList:AddLast(value)
    if 0 == self.count then
        self.count = 1
        self.head = Node.new(self, value)
        self.tail = self.head
        return self.head
    end
    self.count = self.count + 1
    local newTail = Node.new(self, value)
    newTail.prev = self.tail
    self.tail.next = newTail
    self.tail = newTail
    return newTail
end

---移动到链表头
function LinkedList:MoveToFirst(node)
    if node.list ~= self then return end
    if self.head == node then return end
    --断开原来的位置
    if self.tail == node then
        local newTail = node.prev
        newTail.next = nil --断开旧tail
        node.prev = nil --旧tail清除prev
        self.tail = newTail --设置新tail
    else
        node.prev.next = node.next
        node.next.prev = node.prev
        node.prev = nil
    end

    --变成新的head
    node.next = self.head
    self.head.prev = node
    self.head = node
end

function LinkedList:RemoveFirst()
    if 0 == self.count then return nil end

    local ret = self.head
    if 1 == self.count then
        self.head = nil
        self.tail = nil
        self.count = 0
    else
        local newHead = self.head.next
        newHead.prev = nil --断开旧head
        self.head.next = nil--旧head清除next
        self.head = newHead --设置新head
        self.count = self.count - 1
    end
    ret.list = nil
    return ret
end

---移除链表尾
function LinkedList:RemoveLast()
    if 0 == self.count then return nil end

    local ret = self.tail
    if 1 == self.count then
        self.head = nil
        self.tail = nil
        self.count = 0
    else
        local newTail = self.tail.prev
        newTail.next = nil --断开旧tail
        self.tail.prev = nil --旧tail清除prev
        self.tail = newTail --设置新tail
        self.count = self.count - 1
    end
    ret.list = nil
    return ret
end

function LinkedList:Remove(node)
    if node.list ~= self then return end
    if self.head == node then
        self:RemoveFirst()
    elseif self.tail == node then
        self:RemoveLast()
    else
        node.prev.next = node.next
        node.next.prev = node.prev
        node.prev = nil
        node.next = nil
        node.list = nil
    end
end

function LinkedList:__tostring()
    if 0 == self.count then return "" end

    local strTb = {}
    local node = self.head
    while nil ~= node do
        if nil == node.value then
            table.insert(strTb, "nil")
        else
            table.insert(strTb, tostring(node.value))
        end
        node = node.next
    end
    return table.concat(strTb, ",")
end

 

【参考】

LRU算法 - Yrion - 博客园 (cnblogs.com)

 

标签:node,head,return,nil,self,end,LRUCache
来源: https://www.cnblogs.com/sailJs/p/16024264.html