LFU 起夜级李姐

最新推荐文章于 2020-10-29 21:22:44 发布

搬砖党弟中弟

最新推荐文章于 2020-10-29 21:22:44 发布

阅读量228

点赞数

分类专栏： lfu redis 文章标签： redis

本文链接：https://blog.youkuaiyun.com/daima_caigou/article/details/108726046

版权

redis 同时被 2 个专栏收录

16 篇文章

订阅专栏

lfu

1 篇文章

订阅专栏

Go代码实现


type Node struct {
	key, val, freq int
	pre, next      *Node
}


type DoubleLink struct {
	head, tail *Node
	length     int
}


type LFUCache struct {
	freq_map map[int]*DoubleLink
	key_map  map[int]*Node
	cap      int
}


func Constructor(capacity int) LFUCache {
	return LFUCache{
		freq_map: map[int]*DoubleLink{},
		key_map:  map[int]*Node{},
		cap:      capacity,
	}
}

// 初始化双向链表
func InitDoubleLink() *DoubleLink {
	list := &DoubleLink{
		head:   &Node{},
		tail:   &Node{},
		length: 0,
	}
	list.head.next = list.tail
	list.tail.pre = list.head
	return list
}

// 插入双向链表头部
func InsertHead(list *DoubleLink, node *Node) {
	node.next = list.head.next
	list.head.next = node

	node.next.pre = node
	node.pre = list.head

	list.length++
}

// 删除 尾部节点
func RemoveTail(list *DoubleLink) (key int) {
	node := list.tail.pre

	list.tail.pre = node.pre
	node.pre.next = list.tail

	list.length--

	return node.key
}

// 删除节点
func RemoveNode(list *DoubleLink, node *Node) {
	node.pre.next = node.next
	node.next.pre = node.pre

	list.length--
}

func (this *LFUCache) Get(key int) int {
	node, ok := this.key_map[key]
	if !ok {
		return -1
	}
	list := this.freq_map[node.freq]
	RemoveNode(list, node)
	// 频率 +1
	node.freq += 1
	// 把这个 节点 装入 新频率 的 链表头部
	list2, ok := this.freq_map[node.freq]
	if !ok {
		list2 = InitDoubleLink()
		this.freq_map[node.freq] = list2
	}
	InsertHead(list2, node)

	return node.val
}

func (this *LFUCache) Put(key int, value int) {
	if this.cap == 0 {
		return
	}
	node, ok := this.key_map[key]
	// 覆盖旧的value
	if ok {
		// 原来的频率 +1
		list := this.freq_map[node.freq]
		RemoveNode(list, node)
		node.freq++
		node.val = value
		// 把这个 节点 装入 新频率 的 链表头部
		list2, ok := this.freq_map[node.freq]
		if !ok {
			list2 = InitDoubleLink()
			this.freq_map[node.freq] = list2
		}
		InsertHead(list2, node)
	} else {
		// 容量判断 触发淘汰
		if this.cap < len(this.key_map)+1 {
			// 从频率 低到高 的找，看是否存在双向链表
			for i := 1; ; i++ {
				list, ok := this.freq_map[i]
				if ok && list.length > 0 {
					kk := RemoveTail(list)
					delete(this.key_map, kk)
					break
				}
			}
		}
		// 装入keymap
		node = &Node{
			key:  key,
			val:  value,
			freq: 1, // 新增的频率都是1
		}
		this.key_map[key] = node
		// 装入 freq_map 频率=1 的 链表头部
		list, ok := this.freq_map[1]
		if !ok {
			list = InitDoubleLink()
			this.freq_map[1] = list
		}
		InsertHead(list, node)
	}
}