23.设计LRU缓存结构

设计LRU缓存结构

题目描述

    设计LRU缓存结构，该结构在构造时确定大小，假设大小为K，并有如下两个功能
    set(key, value)：将记录(key, value)插入该结构
    get(key)：返回key对应的value值

[要求]
set和get方法的时间复杂度为O(1)
某个key的set或get操作一旦发生，认为这个key的记录成了最常使用的。
当缓存的大小超过K时，移除最不经常使用的记录，即set或get最久远的。
若opt=1，接下来两个整数x, y，表示set(x, y)
若opt=2，接下来一个整数x，表示get(x)，若x未出现过或已被移除，则返回-1
对于每个操作2，输出一个答案
示例1
输入
[[1,1,1],[1,2,2],[1,3,2],[2,1],[1,4,4],[2,2]],3
返回值
[1,-1]
说明
第一次操作后：最常使用的记录为(“1”, 1)
第二次操作后：最常使用的记录为(“2”, 2)，(“1”, 1)变为最不常用的
第三次操作后：最常使用的记录为(“3”, 2)，(“1”, 1)还是最不常用的
第四次操作后：最常用的记录为(“1”, 1)，(“2”, 2)变为最不常用的
第五次操作后：大小超过了3，所以移除此时最不常使用的记录(“2”, 2)，加入记录(“4”, 4)，
并且为最常使用的记录，然后(“3”, 2)变为最不常使用的记录

分析

考虑采用LinkedHashMap来实现，因为它是根据插入顺序来排序的。

代码

import java.util.ArrayList;
import java.util.LinkedHashMap;

public class 设计LRU缓存结构 {
    public int[] LRU (int[][] operators, int k) {
        // write code here
        LinkedHashMap<Integer, Integer> map = new LinkedHashMap<>(); //按插入顺序排序
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 0; i < operators.length; i++) {
            int[] opt = operators[i];
            if(opt[0] == 1){  //set
                //如果要插入的数据的key已经存在过，则要remove()，以便插入新的值后，让新的值成为最常用的值
                if(map.containsKey(opt[1]))
                    map.remove(opt[1]);
                else{
                    if(map.size() == k)  //若大于缓存的大小，则要把最古老的值移除
                        map.remove(map.keySet().toArray()[0]);
                }
                map.put(opt[1], opt[2]);  //把最新的值插入，并成为了最常用的值
            }
            else{    //get
                //如果该key存在，则要把它remove()并保存起来，然后再put进去，以便成为最常用的值
                if(map.containsKey(opt[1])) {
                    Integer value = map.remove(opt[1]);
                    map.put(opt[1], value);
                    list.add(value);
                }
                else list.add(-1);//如果该key不存在，则对LRU结构无影响
            }
        }
        int[] a = new int[list.size()];
        for (int i = 0; i < a.length; i++) {
            a[i] = list.get(i);
        }
        return a;
    }

    public static void main(String[] args) {
        设计LRU缓存结构 d = new 设计LRU缓存结构();
        int[][] operators = {{1,1,1}, {1,2,2}, {1,3,2}, {2,1}, {1,4,4}, {1,3,4}, {2,2}};
        int k = 3;
        int[] a = d.LRU (operators,k);
        for (int i: a) {
            System.out.print(i + " ");
        }
    }
}