java用双向链表加HashMap实现LRU算法

嗯，，，其实在学操作系统的时候用c++实现的了一把LRU算法，即时最近最久未使用嘛。然后前几天面阿里，被面试官问到了这个问题，最近答出来怎么实现，但是实现细节并没有回答的很好，这里用java再去实现一遍，好让自己加深一点点印象，感觉代码的注释已经讲解的很详细了，我就不再多余的赘述了，直接上代码

package cache;

import java.util.HashMap;
/**
 * lru算法其实就是最近最久不使用嘛。
 * 为了实现缓存，就必须用一个map来存储数据达到存取数据的时间复杂度为O(1)
 * 然后为了体现最近最久未使用的思想，我们使用双向链表来实现
 * 总共有四个函数
 * setHead(Object)：我们每次都要把使用的放在双向链表的头部，有可能是不在链表里面的那么就直接放进头部就好了，
 * 如果存在，那就删除链表中的该院书然后在把他放入链表的头部
 * remove(Object,boolean)：主要是一个辅助的函数，主要是删除传入的节点，boolean这个参数的存在就是为了解决如果
 * 缓存满了，那么，我就要去删除tail
 * get(String):根据所提供的key返回value
 * set(String,Object):这个函数主要是将key和value键值对放入mep和链表的
 * @author 11366
 *
 */
public class LRUCache {

	class LRUNode{
		private String key;
		private Object value;
		private LRUNode pre;
		private LRUNode next;
		
		public LRUNode() {
			
		}
		
		public LRUNode(String key,Object value) {
			this.key=key;
			this.value=value;
		}
	}
	
	private HashMap<String, LRUNode> map;
	private LRUNode head;
	private LRUNode tail;
	private int capacity;
	
	public LRUCache() {
		
	}
	
	public LRUCache(int capacity) {
		this.capacity=capacity;
		this.map=new HashMap<>();
	}
	
	public void setHead(LRUNode node) {
		//如果头结点不为空的话，那么就是头插法了
		if(head!=null) {
			node.next=head;
			head.pre=node;
		}
		//无论如何，head始终都是当前的节点node
		head=node;
		//如果尾结点为空，那么就把当前节点赋值给尾结点
		if(tail==null) {
			tail=node;
		}
	}
	
	public void set(String key,LRUNode value) {
		//set是最为关键的一步，我们不但要将节点设置到双向链表当中，我们还需要设置到hashmap当中作为缓存来使用
		//就两个维度吧，，一是将这个节点设置为双向链表的头结点，二是放入hashmap当中
		//先设置为头结点
		LRUNode node = map.get(key);
		setHead(node);
		//然后就是放到hashmap里面去了
		//如果不为空，那么，我们就需要更新他的value
		if(node!=null) {
			node.value=value ;
			//既然不为空，那么我们就需要更新为头结点的啊
			remove(node, false);
		}else {
			node = new LRUNode(key,value);
			if(map.size()>=capacity) {
				remove(tail, true);
			}
			map.put(key, node);
		}
	}
	
	public Object get(String key) {
		LRUNode node = map.get(key);
		//如果存在需要干两件事情，先移出该节点，因为他存在所以我们要移除他在将他放置到头结点的位置
		
		if(node!=null) {
			remove(node, false);
			setHead(node);
			return node.value;
		}
		
		//如果不存在
		return null;
	}
	
	public void remove(LRUNode node,boolean flag) {
		//需要考虑需要删除的节点是head或者是tail的情况
		//如果是head那么就只要让head=node.next就好了，如果是tail那么就只要让tail=node.pre就好了
		if(node.pre!=null) {
			node.pre.next=node.next;
		}else {
			head=node.next;
		}
		if(node.next!=null) {
			node.next.pre=node.pre;
		}else {
			tail=node.pre;
		}
		node.pre=null;
		node.next=null;
		if(flag) {
			map.remove(node.key);
		}
	}
	
}