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

sky_terra

于 2020-07-23 15:09:40 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/sky_terra/article/details/107537268

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

		i = Parent(i)
	}
}

func (r *RankList) Dec