Skip to content

146.LRU 缓存

/*

  • @lc app=leetcode.cn id=146 lang=golang
  • @lcpr version=30204
  • [146] LRU 缓存 */

请你设计并实现一个满足 LRU (最近最少使用) 缓存 约束的数据结构。 实现 LRUCache 类: LRUCache(int capacity) 以 正整数 作为容量 capacity 初始化 LRU 缓存 int get(int key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回 -1 。 void put(int key, int value) 如果关键字 key 已经存在,则变更其数据值 value ;如果不存在,则向缓存中插入该组 key-value 。如果插入操作导致关键字数量超过 capacity ,则应该 逐出 最久未使用的关键字。 函数 get 和 put 必须以 O(1) 的平均时间复杂度运行。

示例:

输入 ["LRUCache", "put", "put", "get", "put", "get", "put", "get", "get", "get"] [[2], [1, 1], [2, 2], [1], [3, 3], [2], [4, 4], [1], [3], [4]] 输出 [null, null, null, 1, null, -1, null, -1, 3, 4]

解释 LRUCache lRUCache = new LRUCache(2); lRUCache.put(1, 1); // 缓存是 {1=1} lRUCache.put(2, 2); // 缓存是 {1=1, 2=2} lRUCache.get(1); // 返回 1 lRUCache.put(3, 3); // 该操作会使得关键字 2 作废,缓存是 {1=1, 3=3} lRUCache.get(2); // 返回 -1 (未找到) lRUCache.put(4, 4); // 该操作会使得关键字 1 作废,缓存是 {4=4, 3=3} lRUCache.get(1); // 返回 -1 (未找到) lRUCache.get(3); // 返回 3 lRUCache.get(4); // 返回 4

提示:

1 <= capacity <= 3000 0 <= key <= 10000 0 <= value <= 105 最多调用 2 * 105 次 get 和 put

数据结构本身没什么难的

go
type LRUCache struct {
	capacity int
	cache    map[int]*list.Element
	lruList  *list.List
}

type entry struct {
	key   int
	value int
}

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

func (this *LRUCache) Get(key int) int {
	if element, ok := this.cache[key]; ok {
		this.lruList.MoveToFront(element) //更新访问时间
		return element.Value.(*entry).value
	}
	return -1
}

func (this *LRUCache) Put(key int, value int) {
	if element, ok := this.cache[key]; ok {
		this.lruList.MoveToFront(element)
		element.Value.(*entry).value = value
	} else {
		if this.lruList.Len() == this.capacity {
			back := this.lruList.Back()
			if back != nil {
				this.lruList.Remove(back)                   //弹出
				delete(this.cache, back.Value.(*entry).key) //删除
			}
		}
		newEntry := &entry{key, value}
		element := this.lruList.PushFront(newEntry)
		this.cache[key] = element
	}
}

/**
 * Your LRUCache object will be instantiated and called as such:
 * obj := Constructor(capacity);
 * param_1 := obj.Get(key);
 * obj.Put(key,value);
 */
本站访客数 人次 本站总访问量