题目来自瓶子君的github
完整题目:
运用你所掌握的数据结构,设计和实现一个 LRU (最近最少使用) 缓存机制。它应该支持以下操作: 获取数据 get 和写入数据 put 。
获取数据 get(key) - 如果密钥 ( key ) 存在于缓存中,则获取密钥的值(总是正数),否则返回 -1 。
写入数据 put(key, value) - 如果密钥不存在,则写入数据。当缓存容量达到上限时,它应该在写入新数据之前删除最久未使用的数据,从而为新数据留出空间。
进阶:
你是否可以在 O(1) 时间复杂度内完成这两种操作?
// 示例
LRUCache cache = new LRUCache( 2 /* 缓存容量 */ );
cache.put(1, 1);
cache.put(2, 2);
cache.get(1); // 返回 1
cache.put(3, 3); // 该操作会使得密钥 2 作废
cache.get(2); // 返回 -1 (未找到)
cache.put(4, 4); // 该操作会使得密钥 1 作废
cache.get(1); // 返回 -1 (未找到)
cache.get(3); // 返回 3
cache.get(4); // 返回 4
实现
class LRUCache {
constructor(maxCache) {
this.maxCache = maxCache; // 最大容量
this.keys = []; // 用来存储
}
put(key, value) {
const index = this.keys.findIndex(item => item.key === key); // 判断是否存在
if (index === -1) {
if (this.keys.length >= this.maxCache) this.keys.splice(0, 1); // 不存在且数组的长度为2了
} else this.keys.splice(index, 1)
// push操作是两边都要进行的,所以直接提出来
this.keys.push({
key,
value
})
}
get(key) {
const index = this.keys.findIndex(item => item.key === key);
if (index === -1) return -1;
else {
const deleteItem = this.keys.splice(index, 1);
this.keys.push(deleteItem[0]);
return deleteItem[0].value;
}
}
}
进阶
// 利用Map的进行操作
class LRUCache {
constructor(maxCache) {
this.maxCache = maxCache;
this.keys = new Map();
}
put(key, value) {
const index = this.keys.has(key); // 判断Map结构里面有没有这个值
if (!index) {
// this.keys.keys().next().value Map.prototype.keys() 返回一个新的 Iterator 对象。
// iterator对象可以使用next函数获得每一项的值,{value: '', done: false | true}
// 所以只需要执行一次next函数就能获取到第一项的key了,然后再把它删掉
if (this.keys.size >= this.maxCache) this.keys.delete(this.keys.keys().next().value);
} else this.keys.delete(key)
this.keys.set(key, value);
}
get(key) {
const index = this.keys.has(key);
if (!index) return -1;
else {
const value = this.keys.get(key)
this.keys.delete(key);
this.keys.set(key, value);
return value;
}
}
}