双向链表+HashMap 实现LRU算法(leetcode 416)

import java.util.HashMap;

/**
 * @author MasterYee
 * @Description:
 * @date: 2020/3/29
 */
public class LRUCache {
    class Node{
        int key;
        int value;
        Node pre;
        Node next;
    }
    public HashMap<Integer,Node> hashMap ;

    public Node head;
    public Node tail;
    public int capacity;
    public int size;

    public LRUCache(int capacity) {
        this.head = new Node();
        this.tail = new Node();
        this.hashMap = new HashMap<>();
        this.head.next = tail;
        this.tail.pre = head;
        this.capacity = capacity;
        this.size = 0;
    }

    // 往头插入node
    public void addNode(Node node){
        node.pre = this.head;
        node.next = this.head.next;

        this.head.next.pre = node;
        this.head.next = node;
    }

    // 把节点放到最前
    public void moveToHead(Node node){
        removeNode(node);
        addNode(node);
    }

    public void removeNode(Node node){
        node.pre.next = node.next;
        node.next.pre = node.pre;
    }

    public Node popTail(){
        Node res = this.tail.pre;
        removeNode(this.tail.pre);
        return res;
    }

    public void put(int key,int value){
        Node node = new Node();
        node.key = key;
        node.value = value;

        Node oldNode = this.hashMap.get(key);
        // 不存在key重复
        if(oldNode == null){
            addNode(node);
            this.hashMap.put(key,node);
            
            // 插入后判断大小 超出则将最老的节点移除
            if(++this.size>capacity){
                Node tail = popTail();
                this.hashMap.remove(tail.key);
                size --;
            }
        }else{
            // 更新值,并设置到头
            oldNode.value = value;
            moveToHead(oldNode);
        }
    }

    public int get(int key){
        Node node = this.hashMap.get(key);
        if(node == null) return -1;

        moveToHead(node);
        return node.value;
    }
}

LRU(Least Recently Used 最近最少使用):常用于缓存更新算法。有一说一缓存大小是固定的,一般是缓存满了后就把最近最老那个资源给删掉,新资源入缓存;每次读取缓存时会刷新当前资源的年龄。

思路

  • 肯定是顺序表实现:每次插入put时往头部put,满了之后将尾部删除,每次get后将当前节点删除、并put到头部,put时若重复了,则覆盖原有值并将其放入头部。
  • 用数组实现:读数能在O(1)时间内,但是缓存满了后再put、和读数后的刷新时间会引起数组结构变化,众所周知数组的增删是O(n)的
  • 用链表实现:保证了新增时和刷新时(删除当前节点、再往头新增当前节点)都是O(1),但是读数慢
  • 双向链表+HashMap:hashmap保证了读取是O(1),链表保证增删是O(1),双向保证不论是头(新增是往头增)还是尾(删除是从尾删)的增删是O(1)
  • 双向链表可以预留一个head和tail,这样不用做判空处理,方便很多
  • 实际上JDK中的LinkedHashMap就能满足要求

这道题据说字节跳动经常考,且要求手撕(不允许用LinkedHashMap,但允许用HashMap,那这就是标准答案)

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值