LRU--LinkedHashMap实现

前言：借助LinkedHashMap，我们可以实现LRU的流程。它内部实现是一个双向链表，是有序的，可以根据数据的访问顺序存储数据，最近访问的数据别放在链表尾部，前提是在初始化的时候指定访问顺序。因此，对应于LRU算法，在查询时，如果命中（访问）就会自动调整到尾部，如果没命中，直接放在尾部即可，若超过指定容量大小，要对头部元素进行移除，这里我们只需要重写LinkedHashMap的removeEldestEntry方法即可。

import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Scanner;

public class LRU {
    static Scanner in = new Scanner(System.in);
	public static void main(String[] args) {
//		LinkedList<> st = new LinkedList<Integer>();
		
		LRUList<Integer,Integer> mp = new LRUList<Integer, Integer>(5, 0.75f, true) ;
		for (int i = 1; i <= 5; i++) {
			mp.put(i, i);
		}
		for (Map.Entry e:mp.entrySet()) {
			System.out.println(e.getKey());
		}
		System.out.println("----------------");
		int key;
		while(in.hasNext()) {
			key = in.nextInt();
			if(mp.containsKey(key)) mp.get(key);
			else mp.put(key, key);
			for (Map.Entry e:mp.entrySet()) {
				System.out.println(e.getKey());
			}
			System.out.println("----------------");
		}
	}

}
class LRUList<K,V> extends LinkedHashMap<K,V>{
	int size;
	public LRUList(int initcap,float loadfactor,boolean order) {
		super(initcap, loadfactor, order);
		this.size = initcap;
	}
     @Override
    protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
    	if(size()>size) return true;
    	else return false;
    } 	
}

拓展：若不能使用这个现成的结构，我们可以自己定义一个方法，使用Hashmap+双向链表的方式即可，Hashmap可以达到O(1)的查找复杂度，稍后更新这部分代码。

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class LRUCache {

    private Map<Integer, ListNode> map;

    /**
     * 双链表结点类
     * 链表头部代表最常访问的节点
     */
    private class ListNode {

        private Integer key;
        private Integer value;

        private ListNode pre;
        private ListNode next;

        public ListNode() {
        }

        public ListNode(Integer key, Integer value) {
            this.key = key;
            this.value = value;
        }
    }

    private int capacity;//容量超过删除

    /**
     * 虚拟头结点没有前驱
     */
    private ListNode dummyHead;
    /**
     * 虚拟尾结点没有后继
     */
    private ListNode dummyTail;

    public LRUCache(int capacity) {
        map = new HashMap<>(capacity);
        this.capacity = capacity;
        dummyHead = new ListNode(-1, -1);
        dummyTail = new ListNode(-1, -1);
        // 初始化链表为 head <-> tail

        dummyHead.next = dummyTail;
        dummyTail.pre = dummyHead;
    }

    /**
     * 命中缓存，将此节点移到链表头部并返回值
     * @param key
     * @return
     */
    public int get(int key) {
        if (map.containsKey(key)) {
            ListNode node = map.get(key);
            int val = node.value;

            // 把当前 node 移动到双向链表的头部
            moveNode2Head(key);
            return val;
        } else {
            return -1;
        }
    }

    /**
     * 发生缺页，则添加新的节点进来
     * 如果哈希表的容量满了，就要删除一个链表末尾元素
     * 然后将新节点在链表头部插入
     *
     * @param key
     * @param value
     */
    public void put(int key, int value) {
        //key 需要更新时使用
        if (map.containsKey(key)) {
            map.get(key).value = value;
            //把当前 node 移动到双向链表的头部
            moveNode2Head(key);
            return;
        }

        // 放元素的操作是一样的

        if (map.size() == capacity) {
            // 如果满了
            ListNode oldTail = removeTail();

            // 设计 key 就是为了在这里删除
            map.remove(oldTail.key);
        }

        // 然后添加元素
        ListNode newNode = new ListNode(key, value);
        map.put(key, newNode);
        addNode2Head(newNode);
    }

    // 为了突出主干逻辑，下面是 3 个公用的方法

    /**
     * 删除双链表尾部结点
     */
    private ListNode removeTail() {
        ListNode oldTail = dummyTail.pre;
        ListNode newTail = oldTail.pre;

        // 两侧结点建立连接
        newTail.next = dummyTail;
        dummyTail.pre = newTail;

        // 释放引用
        oldTail.pre = null;
        oldTail.next = null;

        return oldTail;
    }

    /**
     * 把当前 key 指向的结点移到双向链表的头部
     *
     * @param key
     */
    private void moveNode2Head(int key) {
        // 1、先把 node 拿出来
        ListNode node = map.get(key);

        // 2、原来 node 的前驱和后继接上
        node.pre.next = node.next;
        node.next.pre = node.pre;

        // 3、再把 node 放在末尾
        addNode2Head(node);
    }

    /**
     * 在双链表的头部新增一个结点
     *
     * @param newNode
     */
    private void addNode2Head(ListNode newNode) {
        // 1、当前头结点
        ListNode oldHead = dummyHead.next;

        // 2、末尾结点的后继指向新结点
        oldHead.pre = newNode;

        // 3、设置新结点的前驱和后继
        newNode.pre = dummyHead;
        newNode.next = oldHead;

        // 4、更改虚拟头结点的后继结点
        dummyHead.next = newNode;
    }


    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        LRUCache cache = new LRUCache(3);
        cache.put(1, 1);
        cache.put(2, 2);
        cache.get(2);
        cache.put(3,3);
        cache.put(4,4);
        cache.get(2);
        ListNode head = cache.dummyHead;
        head = head.next;
        while (head!=cache.dummyTail){
            System.out.println(head.value);
            head = head.next;
        }

    }
}