146. LRU 缓存

网站

https://leetcode-cn.com/problems/lru-cache/ (opens new window)

解题思路

• 使用map实现O(1)级别的插入和删除
• get中要判断map是否里面有当前元素，没有的话返回-1，有的话保存value，删除key然后重新添加value并返回
• put中要判断map是否里面有当前元素，有的话，删除key重新添加value
• put中map没有当前元素没有的话，判断是否已经满了，满了要删除之前第一个key，最后无论满不满，都要重新添加value
• PS: 删除第一个key用到了map的迭代器特性，因为map可以保存set的顺序

解题答案

/**
 * @param {number} capacity
 */
var LRUCache = function(capacity) {
    this.size = capacity
    this.map = new Map()
};

/** 
 * @param {number} key
 * @return {number}
 */
LRUCache.prototype.get = function(key) {
    if (this.map.has(key)) {
        const value = this.map.get(key)
        this.map.delete(key)
        this.map.set(key, value)
        return value
    }
    return -1
};

/** 
 * @param {number} key 
 * @param {number} value
 * @return {void}
 */
LRUCache.prototype.put = function(key, value) {
    if (this.map.has(key)) {
        this.map.delete(key)
    } else {
        if (this.map.size >= this.size) {
            this.map.delete(this.map.keys().next().value)
        }
    }
    this.map.set(key, value)
};