【LeetCode-146】 LRU

最新推荐文章于 2024-07-07 20:13:21 发布

原创最新推荐文章于 2024-07-07 20:13:21 发布 · 324 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

leetcode 专栏收录该内容

80 篇文章

订阅专栏

本文介绍了一种使用Java实现LRU（最近最少使用）缓存的方法。通过自定义MyMap类继承LinkedHashMap并覆盖removeEldestEntry方法，可以轻松地实现LRU缓存策略。该方法利用了LinkedHashMap提供的按访问顺序存储的功能。

　　在java中实现LRU的模拟其实很简单，因为提供了现成的数据结构，如果你还不了解，那就一起来学习吧。先把这道题解决了。。。

public class LRUCode {
    MyMap<Integer, Integer> map;

    public LRUCode(int capacity) {
        map = new MyMap<Integer, Integer>(capacity);
    }

    //访问
    public int get(int key) {
        if (!map.containsKey(key)) {
            return -1;
        }

        return map.get(key);
    }

    //添加(这个函数是核心)
    public void set(int key, int value) {
        map.put(key, value);
    }

    public static void main(String[] mh){
        LRUCode lru = new LRUCode(2);
        lru.set(2, 1);
        lru.set(2, 2);
        System.out.println(lru.get(2));
        lru.set(1, 1);
        lru.set(4, 1);
        System.out.println(lru.get(2));
    }

    /**
     * 按访问顺序存储的Map
     *  自己实现map,需要重写removeEldestEntry
     *  removeEldestEntry():
     *      默认情况下返回false : 表示数据永远有效，不会在map中删除
     *      修改后 : 当返回true时，将使最近最久未访问的元素从map中删除
     * 
     * @author MG
     *
     * @param <K>
     * @param <V>
     */
    private static class MyMap<K, V> extends LinkedHashMap<K, V> implements Map<K, V> {

        private static final long serialVersionUID = 1L;

        private int capacity;

        public MyMap(int capacity) {
            super(capacity, 0.75f, true);
            this.capacity = capacity;
        }

        @Override
        protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
            return size() > capacity;
        }
    }
}

　　问题的关键是你必须了解LinkedHashMap这种数据结构，从字面的Link也可以推断这种Map存在指针，没错，它底层就是基本的Map + 双向链表。
　　LinkedHashMap默认情况下按照插入顺序进行存储，即输出的元素顺序和插入的顺序是相同的。它还有另外一个存储顺序，就是安札访问顺序进行存储，说到这里，这不正是LRU嘛，没错，就是的！只要你重写removeEldestEntry这个方法就够了。