golang 最大堆实现排行榜（Top N的排名）

最新推荐文章于 2024-07-31 16:52:17 发布

sky_terra

最新推荐文章于 2024-07-31 16:52:17 发布

阅读量1.8k

点赞数

分类专栏：排行榜 golang 算法文章标签： golang go 算法

本文链接：https://blog.csdn.net/sky_terra/article/details/107537268

版权

本文介绍了如何使用Golang构建最大堆数据结构，并应用其来实现排行榜功能，展示Top N的排名。通过讲解最大堆的插入、删除和调整等操作，详细阐述了在Golang中实现这一算法的过程。

摘要由CSDN通过智能技术生成

package main

import (
	"fmt"
)

/*
 * 使用最大堆实现排行榜功能（前N名排行榜）
 */

type InfoNode struct {
	UserID int
	Score  int
}

type RankList struct {
	MaxHeap  []InfoNode
	HeapSize int

	UserNode map[int]int
}

func NewRankList() *RankList {
	return &RankList{
		UserNode: make(map[int]int),
	}
}

func (r *RankList) UpdateUserScore(userID int, score int) {
	i, exist := r.UserNode[userID]
	if exist && score != r.MaxHeap[i].Score {
		if score > r.MaxHeap[i].Score {
			r.IncreaseScore(i, score)
		} else {
			r.DecreaseScore(i, score)
		}

	}
}

func (r *RankList) IncreaseScore(i int, score int) {
	if score < r.MaxHeap[i].Score {
		return
	}

	r.MaxHeap[i].Score = score
	for i > 0 && r.MaxHeap[Parent(i)].Score < r.MaxHeap[i].Score {
		r.MaxHeap[Parent(i)], r.MaxHeap[i] = r.MaxHeap[i], r.MaxHeap[Parent(i)]
		r.UserNode[r.MaxHeap[Parent(i)].UserID] = Parent(i)
		r.UserNode[r.MaxHeap[i].UserID] = i