本文共 2071 字，大约阅读时间需要 6 分钟。

面云账户时候问了LRU，具体实现的方式是map+双链表。Set和Get的时间复杂度都是O(1)。完整写一遍复习一下, 仅作记录

/** * @Author: lzw5399 * @Date: 2021/5/20 22:28 * @Desc: 基于map和双链表实现的LRU算法 */package mainimport "sync"func main() {	lru := NewLRUCache(3)	lru.Set(1, 233)	lru.Set(2, 666)	lru.Set(3, 777)	lru.Set(5, 888)	lru.Get(2)}// LRUCachetype LRUCache struct {	capacity int	cache    map[int]*LinkedNode	head     *LinkedNode	tail     *LinkedNode	sync.RWMutex}type LinkedNode struct {	key, value int	prev, next *LinkedNode}func NewLRUCache(capacity int) *LRUCache {	return &LRUCache{		capacity: capacity,		cache:    make(map[int]*LinkedNode, capacity),		head:     nil,		tail:     nil,		RWMutex:  sync.RWMutex{},	}}// - key是否已存在//   - 已存在, 将该节点移动到链表头部//   - 未存在, 判断cap是否已满//     - 满//       - 移除链表尾的节点//       - 新的node放入链表头//       - 新的node放入cache的map中//     - 未满//       - 新的node放入链表头//       - 新的node放入cache的map中func (l *LRUCache) Set(key int, value int) {	l.RLock()	node, exist := l.cache[key]	l.RUnlock()	if exist {		l.moveToHead(node)		return	}	node = &LinkedNode{		key:   key,		value: value,	}	l.Lock()	defer l.Unlock()	if l.capacity == len(l.cache) {		removedNode := l.removeTail()		delete(l.cache, removedNode.key)	}	l.addToHead(node)	l.cache[key] = node}// - 从map中获取是否存在//   - 不存在//     - 返回-1//   - 存在//     - 移到链表头部//     - 并返回具体的值func (l *LRUCache) Get(key int) int {	l.RLock()	node, exist := l.cache[key]	l.RUnlock()	if !exist {		return -1	}	l.moveToHead(node)	return node.value}func (l *LRUCache) moveToHead(node *LinkedNode) {	l.removeNode(node)	l.addToHead(node)}func (l *LRUCache) removeTail() *LinkedNode {	return l.removeNode(l.tail)}func (l *LRUCache) removeNode(node *LinkedNode) *LinkedNode {	// 头节点	if node.prev == nil {		l.head = node.next		node.next.prev = nil		return node	}	// 尾节点	if node.next == nil {		l.tail = node.prev		node.prev.next = nil		return node	}	// 中间节点	node.prev.next = node.next	node.next.prev = node.prev	return node}func (l *LRUCache) addToHead(node *LinkedNode) {	if l.head == nil {		l.tail = node	} else {		l.head.prev = node	}	node.prev = nil	node.next = l.head	l.head = node}

转载地址：http://qnkwk.baihongyu.com/