LRU Cache_leetcode_go实现一个LRU缓存,container/list.Remove()内存释放的坑,指针传递,container/list元素改值

LRU Cache_leetcode_go实现一个LRU缓存,container/list.Remove()内存释放的坑,指针传递,container/list元素改值

题目:

Design and implement a data structure for Least Recently Used (LRU) cache. It should support the following operations: get and put.

get(key) - Get the value (will always be positive) of the key if the key exists in the cache, otherwise return -1.
put(key, value) - Set or insert the value if the key is not already present. When the cache reached its capacity, it should invalidate the least recently used item before inserting a new item.

Follow up:
Could you do both operations in O(1) time complexity?

Example:

LRUCache cache = new LRUCache( 2 /* capacity */ );

cache.put(1, 1);
cache.put(2, 2);
cache.get(1);       // returns 1
cache.put(3, 3);    // evicts key 2
cache.get(2);       // returns -1 (not found)
cache.put(4, 4);    // evicts key 1
cache.get(1);       // returns -1 (not found)
cache.get(3);       // returns 3
cache.get(4);       // returns 4

思路:

  • 一道设计数据结构的题,不涉及算法,细节多。同时维护哈希表和链表保存数据。哈希实现快速查找,链表记录LRU。每次Put/Get都把节点挪到LRU链表的最后(先删后加)。
  • 细节1:可选的面向对象编程,减少形参传递。
  • 细节2:自己写双链表,或者用container/list。container/list本身就是双链表实现。root.prev指向尾节点,所以找头尾节点都不需要遍历。list.Front(),list.Back()执行时间均为O(1)。
  • 细节3:cap指的是LRU容量,Get不需要修改,Put时超过容量,则踢除最少使用的节点。这也是用双链表的原因。
  • 细节4:为什么map的v要用*list.Ellement。

错解(存在无法更新值的问题):

type LRUCache struct {
    lru   *list.List
    store map[int]*list.Element
    cap   int
}

type LruKv struct {
    lruK int
    lruV int
}

func Constructor(capacity int) LRUCache {
    return LRUCache{
        lru:   list.New(),
        store: make(map[int]*list.Element, capacity),
        cap:   capacity,
    }
}

func (this *LRUCache) Get(key int) int {
    elem, ok := this.store[key]
    if !ok {
        return -1
    }

    //更新lru
    this.lru.Remove(elem) //map保存指针,方便删除
    node := elem.Value.(LruKv)
    this.lru.PushBack(LruKv{lruK: node.lruK, lruV: node.lruV,})
    
    return elem.Value.(LruKv).lruV
}

func (this *LRUCache) Put(key int, value int) {
    elem, ok := this.store[key]
    if ok {
        lruKvObj := elem.Value.(LruKv)
        lruKvObj.lruV = value //map的v存成指针,方便更新值
        this.lru.Remove(elem)
        this.lru.PushBack(elem.Value.(LruKv))
    } else {
        this.lru.PushBack(LruKv{lruK:key,lruV:value})
        lruKvObj := this.lru.Back()
        this.store[key] = lruKvObj
    }

    //处理cap
    if this.lru.Len() > this.cap {
        elem := this.lru.Front()
        node := elem.Value.(LruKv)
        delete(this.store, node.lruK)
        this.lru.Remove(elem)
    }
}

解决:要注意container/list中的elem.Value.(type)是值传递,而不是引用传递,无法改值。只重重建节点再插入。解决办法是改list.Element.Value的类型为*LruKv。

错解:(有一个关于container/list修改值的坑)

package main

import (
    "container/list"
)

type LRUCache struct {
    lru   *list.List
    store map[int]*list.Element
    cap   int
}

type LruKv struct {
    lruK int
    lruV int
}

func Constructor(capacity int) LRUCache {
    return LRUCache{
        lru:   list.New(),
        store: make(map[int]*list.Element, capacity),
        cap:   capacity,
    }
}

func (this *LRUCache) Get(key int) int {
    elem, ok := this.store[key]
    if !ok {
        return -1
    }

    //更新lru
    this.lru.Remove(elem) //map保存指针,方便删除
    node := elem.Value.(*LruKv)
    this.lru.PushBack(&LruKv{lruK: node.lruK, lruV: node.lruV,})

    return elem.Value.(*LruKv).lruV
}

func (this *LRUCache) Put(key int, value int) {
    elem, ok := this.store[key]
    if ok {
        lruKvObj := elem.Value.(*LruKv)
        lruKvObj.lruV = value //map的v存成指针,方便更新值
        this.lru.Remove(elem)
        this.lru.PushBack(lruKvObj)
    } else {
        this.lru.PushBack(&LruKv{lruK:key,lruV:value})
        lruKvObj := this.lru.Back()
        this.store[key] = lruKvObj
    }

    //处理cap
    if this.lru.Len() > this.cap {
        elem := this.lru.Front()
        node := elem.Value.(*LruKv)
        delete(this.store, node.lruK)
        this.lru.Remove(elem)
    }
}

问题分析:一个用container/list的坑,this.lru.Remove(e Element)。为避免内存泄露,释放了e的存储空间。实现为利用e的prev next指针进行删除。所以要求Element不仅要有Value,还要附带链表信息。
由于是引用传递,所以会清空map[]中的值。导致出现的问题是map[key]中的*Element仅有Value,没有链表信息。在插入重复元素时删除会失败。产生了冗余元素。

所以矛盾冲突在于list.Remove()方法的形参是引用传递,而本例实现map也是引用传递。在调用list.Remove()时,会析构形参*Element,影响其作为map的value被使用。影响的结果是清空了*Element所属的链表信息。进而导致更新value时,执行失败。元素个数多于预期。进而触发了扩容清理,从而导致逻辑错误。

解决:调用l.Remove()之后,更新map[]的对应值。

package main

import (
    "container/list"
    "fmt"
)

type LRUCache struct {
    lru   *list.List
    store map[int]*list.Element
    cap   int
}

type LruKv struct {
    lruK int
    lruV int
}

func Constructor(capacity int) LRUCache {
    return LRUCache{
        lru:   list.New(),
        store: make(map[int]*list.Element, capacity),
        cap:   capacity,
    }
}

func (this *LRUCache) Get(key int) int {
    elem, ok := this.store[key]
    if !ok {
        return -1
    }

    //更新lru
    this.lru.Remove(elem) //map保存指针,方便删除
    node := elem.Value.(*LruKv)
    this.lru.PushBack(&LruKv{lruK: node.lruK, lruV: node.lruV,})
    this.store[key] = this.lru.Back()

    return elem.Value.(*LruKv).lruV
}

func (this *LRUCache) Put(key int, value int) {
    elem, ok := this.store[key]
    if ok {
        lruKvObj := elem.Value.(*LruKv)
        lruKvObj.lruV = value //map的v存成指针,方便更新值
        this.lru.Remove(elem)
        this.lru.PushBack(lruKvObj)
        //特殊应对l.Remove()
        this.store[key] = this.lru.Back()
    } else {
        this.lru.PushBack(&LruKv{lruK:key,lruV:value})
        lruKvObj := this.lru.Back()
        this.store[key] = lruKvObj
    }

    //处理cap
    if this.lru.Len() > this.cap {
        elem := this.lru.Front()
        node := elem.Value.(*LruKv)
        delete(this.store, node.lruK)
        this.lru.Remove(elem)
    }
}

这个题测试输入太长,并且链表的变量跟踪进入的很深。调试了半天,所以mark一下Accepted。还有需要改进的地方。附注*Element存入map是指针,*Element.Value的值也是指针,才能实现改值。要不就要重新构建,并给map赋值。这也带了上述的问题。

Runtime: 176 ms, faster than 75.26% of Go online submissions for LRU Cache.

//TODO 需要做的优化,直接用双链表实现。container/list的Element很难用,编程变量都是一大串。还要做类型判断,简便地不要用.(type),直接用.(LruKv)一行代码返回对象。

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