快手一面 手写LRU

import java.util.HashMap;
public class LRUCache {
    //循环双向链表
    class DLinkedNode{
        int key;
        int value;
        DLinkedNode prev;
        DLinkedNode next;
        public DLinkedNode(){}
        public DLinkedNode(int key,int value){
            this.key = key;
            this.value = value;
        }
    }
    private HashMap<Integer,DLinkedNode> map = new HashMap<>();
    private int size; //当前的大小
    private int cap; //表示容量
    private DLinkedNode head,tail; //使用一个虚拟头和虚拟尾节点 判断的时候容易处理

    public LRUCache(int cap){
        this.size = 0;
        this.cap = cap;
        head = new DLinkedNode();
        tail = new DLinkedNode();
        head.next = tail;
        tail.next = head;
    }
    public int get(int key){
        DLinkedNode node = map.get(key);
        if(node == null){
            return -1;
        }
        //如果存在 将当前节点移动到头部
        moveCurToHead(node);
        
        return node.value;
    }
    public void put(int key,int value){
        DLinkedNode node = map.get(key);
        if(node == null){
            //表示set的key不存在 则新建节点插入头部
            DLinkedNode newNode = new DLinkedNode(key, value);
            map.put(key,newNode);
            size++;
            addNodeToHead(newNode);
            //这时需要判断当前的长度是否大于最大容量 如果大于就需要把最后一个节点移除
            if(size > cap){
                DLinkedNode tail = removeTail();
                map.remove(tail.key);
                size--;
            }
        }else{
            //否则 修改值 并移动到头部
            node.value = value;
            moveCurToHead(node);
        }
    }
    public DLinkedNode removeTail() {
        //这里有一个虚拟尾部节点
       DLinkedNode node = tail.prev;
       removeNode(node);
        return node;
    }
    public void moveCurToHead(DLinkedNode node) {
        removeNode(node);
        addNodeToHead(node);
    }
    public void addNodeToHead(DLinkedNode newNode){
        newNode.next = head.next;
        newNode.prev = head;
        head.next.prev = newNode;
        head.next = newNode;
    }
    public void removeNode(DLinkedNode node){
        node.prev.next = node.next;
        node.next.prev = node.prev;
    }
    public static void main(String[] args) {
        LRUCache lru = new LRUCache(4);
        // 1 2 3 4 1 5  应该是2没有
        lru.put(1, 1);
        lru.put(2, 2);
        lru.put(3, 3);
        lru.put(4, 4);
        System.out.println(lru.get(1));
        lru.put(5, 5);
        System.out.println(lru.get(3));
    }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值