设计LRU缓存结构

设计LRU缓存结构

题目描述

设计LRU缓存结构，该结构在构造时确定大小，假设大小为K，并有如下两个功能

• set(key, value)：将记录(key, value)插入该结构
• get(key)：返回key对应的value值

[要求]

1. set和get方法的时间复杂度为O(1)
2. 某个key的set或get操作一旦发生，认为这个key的记录成了最常使用的。
3. 当缓存的大小超过K时，移除最不经常使用的记录，即set或get最久远的。

若opt=1，接下来两个整数x, y，表示set(x, y)
若opt=2，接下来一个整数x，表示get(x)，若x未出现过或已被移除，则返回-1
对于每个操作2，输出一个答案

解题思路

使用LinkedHashMap数据结构，最先进去的排在链头

代码

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> lruM = new LinkedHashMap<Integer, Integer>();
        ArrayList<Integer> resTemp = new ArrayList<Integer>();
        for(int[] op: operators){
            switch(op[0]){
                case 1:
                    if(lruM.size()>=k){
                        Iterator dd = lruM.keySet().iterator();
                        lruM.remove(dd.next());
                        lruM.put(op[1],op[2]);
                    }else{
                        lruM.put(op[1],op[2]);
                    }
                    break;
                case 2:
                    if(lruM.containsKey(op[1])){
                        int temp = lruM.get(op[1]);
                        lruM.remove(op[1]);
                        lruM.put(op[1], temp);
                        resTemp.add(temp);
                    }else{
                        resTemp.add(-1);
                    }
                    break;
                default:;
            }
        }
        int[] resArray = new int[resTemp.size()];
        int i = 0;
        for(int res:resTemp){
            resArray[i++] = res;
        }
        return resArray;
    }
}