LinkedHashMap 简单实现LRU

骑猪看日落 2024-02-21 11:05 62阅读 0赞

要使用 `LinkedHashMap` 来实现LRU（最近最少使用）缓存，可以设置它的访问顺序为`true`，以便在每次访问一个元素时，将它移到最后，从而实现LRU的特性。以下是一个简单的Java示例：

import java.util.LinkedHashMap;
    import java.util.Map;
    
    public class LRUCache<K, V> extends LinkedHashMap<K, V> {
        
        private final int MAX_CAPACITY;
    
        public LRUCache(int capacity) {
        
            super(capacity, 0.75f, true);
            MAX_CAPACITY = capacity;
        }
    
        @Override
        protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
        
            return size() > MAX_CAPACITY;
        }
    
        public static void main(String[] args) {
        
            LRUCache<Integer, String> lruCache = new LRUCache<>(3);
    
            lruCache.put(1, "One");
            lruCache.put(2, "Two");
            lruCache.put(3, "Three");
    
            System.out.println(lruCache); // 输出: {1=One, 2=Two, 3=Three}
    
            lruCache.get(2); // 访问元素2
    
            System.out.println(lruCache); // 输出: {1=One, 3=Three, 2=Two}
    
            lruCache.put(4, "Four"); // 添加一个新元素
    
            // 1最先放入
            System.out.println(lruCache); // 输出: {3=Three, 2=Two, 4=Four}
    
            lruCache.put(5, "Five"); // 添加一个新元素
    
            // 2先放入但使用了一次，3紧随其后但没有使用
            System.out.println(lruCache); // 输出: {2=Two, 4=Four, 5=Five}
        }
    }

在上面的示例中，创建了一个 `LRUCache` 类，它扩展了 `LinkedHashMap`，并在构造函数中设置了访问顺序为`true`，以便实现LRU缓存。重写了 `removeEldestEntry` 方法，以控制缓存的大小，如果缓存大小超过最大容量（`MAX_CAPACITY`），则会移除最近最少使用的元素。