LRU-Swift实现

双向链表+Map实现,get、put、时间复杂度为O(1).
LRU数据结构如下图:


LRU数据结构,该图来自https://marcosantadev.com/implement-cache-lru-swift/#cache_lru

LRU

LRU(least recently used) 最近最少使用的 ,核心思想:如果数据最近被访问,将来被访问的概率也高

步骤:

1.新插入数据放在表头
2.最近访问数据移动到表头
3.当链表满了,就将尾部数据丢弃

代码实现

// MARK: 输出打印
protocol Printable {
    func reversePrint()
    func printf()
}

/// 链表节点
class LinkNode<T> {
    var value: T
    var next: LinkNode? = nil // 下一个节点
    weak var previous: LinkNode? = nil // 前一个节点
    init(_ value: T) {
        self.value = value
    }
}

// MARK: LRU缓存实现(双向链表+map)
final class LRUCache<Key, Value> {
    private struct CachePaylaod {
        var key: Key
        var value: Value
    }
    
    private typealias Node = LinkNode<CachePaylaod>
    private var _head: Node?
    private var _tail: Node?
    private var _count: Int = 0
    var count: Int {
        return _count
    }
    
    private var _initCapacity: Int
    
    private var _lruMap: [Key: LinkNode<CachePaylaod>]
    
    init(_ capacity: Int) {
        self._initCapacity = max(0, capacity)
        
        _lruMap = [Key: LinkNode<CachePaylaod>].init(minimumCapacity: capacity)
    }
    
    /// 取值
    func get(_ key: Key) -> Value? {
        if let node = self._lruMap[key] {
            self.moveNodeToHead(node: node)
        }
        return self._lruMap[key]?.value.value
    }
    
    /// 设置值
    func put(_ key: Key, _ value: Value) {
        let cachePayload = CachePaylaod.init(key: key, value: value)
        /*
        1、key 存在,更新value,并将当前节点移到链表头部
        2、key不存在,创建新的节点并拼接到头部
        3、超过最大容量,移除最后一个节点
        */
        if let node = self._lruMap[key] {
            node.value = cachePayload
            moveNodeToHead(node: node)
        }else {
            let node = LinkNode.init(cachePayload)
            appenToHead(newNode: node)
            _lruMap[key] = node
        }
        
        if _count > _initCapacity {
            let nodeRemove = removeLastNode()
            if let key = nodeRemove?.value.key {
                _lruMap[key] = nil
            }
        }
    }
    
    /// 将存在的节点移到头部
    private func moveNodeToHead(node: Node) {
        if node === _head {
            return
        }
        let next = node.next
        let previous = node.previous
        if previous === _head {
            previous?.previous = node
        }
        if node === _tail {
            _tail = previous
        }
        previous?.next = next
        next?.previous = previous
        
        node.previous = nil
        node.next = _head
        _head?.previous = node
        _head = node
    }
    
    
    /// 拼接一个新的节点到头部
    private func appenToHead(newNode: Node) {
        if _head == nil {
            _head = newNode
            _tail = _head
        }else{
            let temp = newNode
            temp.next = _head
            _head?.previous = temp
            _head = temp
        }
        _count += 1

    }
    
    /// 移除最后一个节点
    @discardableResult
    private func removeLastNode() -> Node? {
        if let tail = _tail {
            let tailPre = tail.previous
            tail.previous = nil
            tailPre?.next = nil
            _tail = tailPre
            if _count == 1 {
                _head = nil
            }
            _count -= 1
            return tail
        }else {
            return nil
        }

    }
}

extension LRUCache: Printable {
    func reversePrint() {
        var node = _tail
        var ll = [Value]()
        while node != nil {
            ll.append(node!.value.value)
            node = node?.previous
        }
        print(ll)
    }
    
    func printf() {
        var node = _head
        var ll = [Value]()
        while node != nil {
            ll.append(node!.value.value)
            node = node?.next
        }
        print(ll)
    }
}

测试:

let lru = LRUCache<Int, Int>.init(3)
lru.put(1, 1)
lru.put(2, 2)
lru.printf()            打印结果:[2, 1]
lru.reversePrint()      打印结果:[1, 2]
lru.get(1)            
lru.printf()            打印结果:[1, 2]
lru.reversePrint()      打印结果:[2, 1]

lru.put(3, 3)
lru.printf()            打印结果:[3, 1, 2]
lru.reversePrint()      打印结果:[2, 1, 3]
lru.get(2)              
lru.printf()            打印结果:[2, 3, 1]
lru.reversePrint()      打印结果:[1, 3, 2]
lru.put(4, 4)
lru.printf()            打印结果:[4, 2, 3]
lru.reversePrint()      打印结果:[3, 2, 4]
lru.get(1)              结果为nil
lru.printf()            打印结果:
lru.reversePrint()      打印结果:
lru.get(3)
lru.printf()            打印结果:[3, 4, 2]
lru.reversePrint()      打印结果:[2, 4, 3]
lru.get(4)
lru.printf()            打印结果:[4, 3, 2]
lru.reversePrint()      打印结果:[2, 3, 4]
lru.put(3, 30)
lru.printf()            打印结果:[30, 4, 2]
lru.reversePrint()      打印结果:[2, 4, 30]

demo链接

题外话

该代码实现leetcode居然编译报错😤。不知道是不是泛型的影响。
虽然得到权威的认证,不过我坚信这个实现没问题😏。
当然,如有问题欢迎各路大神指点🤪。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,332评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,508评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,812评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,607评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,728评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,919评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,071评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,802评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,256评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,576评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,712评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,389评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,032评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,798评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,026评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,473评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,606评论 2 350

推荐阅读更多精彩内容