緩存算法——LRU算法編碼實(shí)現(xiàn)(Java版)

LRU算法(Last Recently Used)题涨,是緩存淘汰算法的其中一種,即最近最久未使用锄贼。

在操作系統(tǒng)的內(nèi)存管理中曲稼,有一類很重要的算法就是內(nèi)存頁(yè)面置換算法(包括FIFO榴都,LRU,LFU等幾種常見頁(yè)面置換算法)待锈。

事實(shí)上,Cache算法和內(nèi)存頁(yè)面置換算法的核心思想是一樣的:都是在給定一個(gè)限定大小的空間的前提下嘴高,設(shè)計(jì)一個(gè)原則如何來(lái)更新和訪問(wèn)其中的元素炉擅。

下面說(shuō)一下LRU算法的核心思想辉懒,LRU算法的設(shè)計(jì)原則是:
如果一個(gè)數(shù)據(jù)在最近一段時(shí)間沒有被訪問(wèn)到,那么在將來(lái)它被訪問(wèn)的可能性也很小谍失。也就是說(shuō)眶俩,當(dāng)限定的空間已存滿數(shù)據(jù)時(shí),應(yīng)當(dāng)把最久沒有被訪問(wèn)到的數(shù)據(jù)淘汰

這里提供兩種使用Java的實(shí)現(xiàn)方式



使用LinkedHashMap

具體實(shí)現(xiàn)源碼:

public class LRUCache<K, V> extends LinkedHashMap<K, V> {

    private final int CACHE_SIZE;


    /**
     * 構(gòu)造器快鱼,構(gòu)造緩存的大小颠印,即最多能緩存多少參數(shù)
     *
     * 初始化的 initialCapacity 結(jié)果+1,是為了在cacheSize小于0.75時(shí)除的結(jié)果為0抹竹,但是這時(shí)候要有能存儲(chǔ)的元素线罕,所以+1
     * accessOrder:控制訪問(wèn)順序。用于設(shè)置LinkedHashMap在調(diào)用 get() 之后的操作窃判。如果設(shè)置為true钞楼,則每次調(diào)用get后會(huì)將該元素移動(dòng)到末尾
     * removeEldestEntry:默認(rèn)返回為false,在調(diào)用 afterNodeInsertion()的內(nèi)部袄琳,調(diào)用該方法询件。也就是說(shuō),每次添加元素之后唆樊,檢查并移除最后一個(gè)元素
     *
     *
     * @param cacheSize
     */
    public LRUCache(int cacheSize) {
        super((int) Math.ceil(cacheSize / 0.75)+1, 0.75F, true);
        CACHE_SIZE = cacheSize;
    }

    /**
     * 重寫removeEldestEntry
     * 當(dāng)map中的數(shù)據(jù)大于指定緩存大小的時(shí)候宛琅,就刪除最老的數(shù)據(jù)
     * @param eldest
     * @return
     */
    @Override
    protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
        return size() > CACHE_SIZE;
    }
}

LinkedHashMap中關(guān)鍵方法的源碼:

    // 添加Node之后的操作,可以看到調(diào)用了 removeEldestEntry逗旁,移除頭部元素
    void afterNodeInsertion(boolean evict) { // possibly remove eldest
        LinkedHashMap.Entry<K,V> first;
        if (evict && (first = head) != null && removeEldestEntry(first)) {
            K key = first.key;
            removeNode(hash(key), key, null, false, true);
        }
    }

    // 訪問(wèn)Node之后的操作嘿辟,可以看到,當(dāng)設(shè)置了accessOrder以后片效,訪問(wèn)元素之后红伦,會(huì)把該Node放到尾部
    void afterNodeAccess(Node<K,V> e) { // move node to last
        LinkedHashMap.Entry<K,V> last;
        if (accessOrder && (last = tail) != e) {
            LinkedHashMap.Entry<K,V> p =
                (LinkedHashMap.Entry<K,V>)e, b = p.before, a = p.after;
            p.after = null;
            if (b == null)
                head = a;
            else
                b.after = a;
            if (a != null)
                a.before = b;
            else
                last = b;
            if (last == null)
                head = p;
            else {
                p.before = last;
                last.after = p;
            }
            tail = p;
            ++modCount;
        }
    }

可以看到,LinkedHashMap事實(shí)上已經(jīng)提供了內(nèi)存置換的功能淀衣,我們利用了其中內(nèi)置的功能并重寫了部分方法后色建,實(shí)現(xiàn)了LRUCache。

需要注意兩點(diǎn)
1舌缤、構(gòu)造器中調(diào)用父方法的第三個(gè)參數(shù) accessOrder(默認(rèn)為false),通過(guò)源碼注釋可以知道某残,這個(gè)表示元素的訪問(wèn)模式国撵。true為access-order,false為insertion-order
當(dāng)accessOrder=true時(shí)玻墅,用于搭配方法 removeEldestEntry(Map.Entry<K, V> eldest) 使用介牙。

2、受保護(hù)方法 removeEldestEntry(Map.Entry<K, V> eldest)澳厢,用于判斷是否移除最老一個(gè)元素环础。使得Cache可以在元素滿之后移除最老的元素囚似。



使用HashMap 編碼實(shí)現(xiàn)LRUCache

具體實(shí)現(xiàn)源碼:

public class LRUCacheCustom<K,V> {

    // 雙鏈表
    static class CacheNode{
        CacheNode before;
        CacheNode after;
        Object key;
        Object val;
        public CacheNode(){}
    }

    /** 頭節(jié)點(diǎn) */
    private CacheNode head;
    /** 尾節(jié)點(diǎn) */
    private CacheNode tail;

    /** 最大元素?cái)?shù) */
    private int maxCapacity;

    /** 用于存儲(chǔ)節(jié)點(diǎn)數(shù)據(jù) */
    private HashMap<K, CacheNode> caches;


    /** 初始化 */
    public LRUCacheCustom(int maxCapacity){
        this.maxCapacity = maxCapacity;
        caches = new HashMap<>(maxCapacity);
        head = new CacheNode();
        tail = new CacheNode();
        head.after = tail;
        tail.before = head;
    }


    public void put(K k, V v){
        CacheNode node = caches.get(k);
        if(node == null){
            node = new CacheNode();
            node.key = k;
        }

        node.val = v;
        moveToFirst(node);
        caches.put(k, node);

        if(caches.size() > maxCapacity){
            caches.remove(tail.before.key);
            removeLast();
        }
    }


    public V get(K k){
        CacheNode node = caches.get(k);
        if(node == null){
            return null;
        }
        Object val = node.val;
        moveToFirst(node);
        return (V) val;
    }


    public void removeLast(){
        CacheNode last = tail.before;
        CacheNode secondLast = tail.before.before;
        secondLast.after = tail;
        tail.before = secondLast;

        // 手動(dòng)釋放內(nèi)存
        caches.remove(last.key);
        last = null;
    }

    public void moveToFirst(CacheNode node){
        // 先把原來(lái)node的前后連起來(lái),如果前后不為null的話线得。(注意順序不能錯(cuò))
        CacheNode nodeBefore = node.before;
        if(node.after != null){
            node.after.before = nodeBefore;
        }
        if(nodeBefore != null ){
            nodeBefore.after = node.after;
        }

        // 然后把node放前面
        CacheNode tmp = head.after;
        head.after = node;
        node.before = head;
        tmp.before = node;
        node.after = tmp;
    }


    @Override
    public String toString() {
        return caches.toString();
    }

    public String getCache(){
        StringBuilder bu = new StringBuilder();
        CacheNode dummy = head;
        while(dummy.after != null){
            bu.append("{key=" + dummy.key);
            bu.append(", val=" + dummy.val);
            bu.append("\n -->");
            dummy = dummy.after;
        }
        return bu.toString();
    }
}

以上代碼我已經(jīng)自測(cè)通過(guò)
測(cè)試代碼:

    @Test
    public void lruCacheTest(){
        LRUCacheCustom cacheCustom = new LRUCacheCustom(3);
        cacheCustom.put(1,"AAA");
        cacheCustom.put(2,"BBB");
        cacheCustom.put(3,"CCC");
        System.out.println(cacheCustom.getCache()); // 3 2 1

        cacheCustom.put(4,"DDD");
        System.out.println(cacheCustom.getCache()); // 4 3 2

        // get(2) 將2提升到頂部
        cacheCustom.get(2);
        System.out.println(cacheCustom.getCache()); // 2 4 3

        cacheCustom.put(5,"DDD");
        System.out.println(cacheCustom.getCache()); // 5 2 4
    }


參考:
LRU Cache

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末饶唤,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子贯钩,更是在濱河造成了極大的恐慌募狂,老刑警劉巖,帶你破解...
    沈念sama閱讀 206,311評(píng)論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件角雷,死亡現(xiàn)場(chǎng)離奇詭異祸穷,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)勺三,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,339評(píng)論 2 382
  • 文/潘曉璐 我一進(jìn)店門雷滚,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人吗坚,你說(shuō)我怎么就攤上這事祈远。” “怎么了刻蚯?”我有些...
    開封第一講書人閱讀 152,671評(píng)論 0 342
  • 文/不壞的土叔 我叫張陵绊含,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我炊汹,道長(zhǎng)躬充,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 55,252評(píng)論 1 279
  • 正文 為了忘掉前任讨便,我火速辦了婚禮充甚,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘霸褒。我一直安慰自己伴找,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,253評(píng)論 5 371
  • 文/花漫 我一把揭開白布废菱。 她就那樣靜靜地躺著技矮,像睡著了一般。 火紅的嫁衣襯著肌膚如雪殊轴。 梳的紋絲不亂的頭發(fā)上衰倦,一...
    開封第一講書人閱讀 49,031評(píng)論 1 285
  • 那天,我揣著相機(jī)與錄音旁理,去河邊找鬼樊零。 笑死,一個(gè)胖子當(dāng)著我的面吹牛孽文,可吹牛的內(nèi)容都是我干的驻襟。 我是一名探鬼主播夺艰,決...
    沈念sama閱讀 38,340評(píng)論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼沉衣!你這毒婦竟也來(lái)了郁副?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 36,973評(píng)論 0 259
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤厢蒜,失蹤者是張志新(化名)和其女友劉穎霞势,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體斑鸦,經(jīng)...
    沈念sama閱讀 43,466評(píng)論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡愕贡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,937評(píng)論 2 323
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了巷屿。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片固以。...
    茶點(diǎn)故事閱讀 38,039評(píng)論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖嘱巾,靈堂內(nèi)的尸體忽然破棺而出憨琳,到底是詐尸還是另有隱情,我是刑警寧澤旬昭,帶...
    沈念sama閱讀 33,701評(píng)論 4 323
  • 正文 年R本政府宣布篙螟,位于F島的核電站,受9級(jí)特大地震影響问拘,放射性物質(zhì)發(fā)生泄漏遍略。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,254評(píng)論 3 307
  • 文/蒙蒙 一骤坐、第九天 我趴在偏房一處隱蔽的房頂上張望绪杏。 院中可真熱鬧,春花似錦纽绍、人聲如沸蕾久。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,259評(píng)論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)僧著。三九已至,卻和暖如春障簿,著一層夾襖步出監(jiān)牢的瞬間盹愚,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,485評(píng)論 1 262
  • 我被黑心中介騙來(lái)泰國(guó)打工卷谈, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人霞篡。 一個(gè)月前我還...
    沈念sama閱讀 45,497評(píng)論 2 354
  • 正文 我出身青樓世蔗,卻偏偏與公主長(zhǎng)得像端逼,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子污淋,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,786評(píng)論 2 345

推薦閱讀更多精彩內(nèi)容