牛客网、设计LRU缓存结构

题目描述

关于LRU算法和LFU算法：
知乎：LRU缓存结构

解决办法

使用内置的数据：LinkedHashMap，其由hashmap和链表，链表负责维护最常使用和最久不使用的数据。该链表为双向链表，可以很快的移除头节点和尾节点。同时该链表还必须存储key-val，方便删除时，得到删除节点的key，在map里面移除该key。

import java.util.*;

public class Solution {
    /**
     * lru design
     * @param operators int整型二维数组 the ops
     * @param k int整型 the k
     * @return int整型一维数组
     */
    public int[] LRU (int[][] operators, int k) {
        // write code here
        LinkedHashMap<Integer,Integer> lruMap = new LinkedHashMap<Integer,Integer>();
        ArrayList<Integer> result = new ArrayList();
        
        for(int[] opat:operators){
            int key = opat[1];
            switch(opat[0]){
                case 1:
                    int value = opat[2];
                    if(lruMap.size()<k){
                        lruMap.put(key,value);
                    }else{
                        Iterator ot = lruMap.keySet().iterator();
                        lruMap.remove(ot.next());//移除最不长使用的
                        lruMap.put(key,value);//添加，头插法
                    }
                    break;
                case 2:
                    if(lruMap.containsKey(key)){
                        int val = lruMap.get(key);
                        result.add(val);
                        lruMap.remove(key);//因为这个时候key使用了，变为最常使用了
                        lruMap.put(key,val);//添加到头
                    }else{
                        result.add(-1);
                    }
                    break;
                default:
            }
        }
        int[] resultArr = new int[result.size()];
            int i=0;
            for(int a:result){
                resultArr[i++]=a;
            }
        return resultArr;
    }
}