运用你所掌握的数据结构,设计和实现一个 LRU (最近最少使用) 缓存机制 。
实现 LRUCache 类:
LRUCache(int capacity) 以正整数作为容量 capacity 初始化 LRU 缓存
int get(int key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回 -1 。
void put(int key, int value) 如果关键字已经存在,则变更其数据值;如果关键字不存在,则插入该组「关键字-值」。当缓存容量达到上限时,它应该在写入新数据之前删除最久未使用的数据值,从而为新的数据值留出空间。
进阶:你是否可以在 O(1) 时间复杂度内完成这两种操作?
class LRUCache { public int cap; public Map<Integer, Integer> map = new LinkedHashMap<Integer, Integer>(); public LRUCache(int capacity) { this.cap = capacity; } public int get(int key) { if (map.containsKey(key)) { int value = map.get(key); map.remove(key); map.put(key,value); return value; } return -1; } public void put(int key, int value) { if (map.containsKey(key)) { map.remove(key); } else if (map.size() == cap) { Iterator<Integer> iterator = map.keySet().iterator(); iterator.next(); iterator.remove(); } map.put(key, value); } }