Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4963780a32 | ||
|
|
70ebaae3d2 | ||
|
|
9cd286fd92 |
52
tools/random.go
Normal file
52
tools/random.go
Normal file
@@ -0,0 +1,52 @@
|
||||
package tools
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
Rand *randMod
|
||||
)
|
||||
|
||||
type randMod struct {
|
||||
}
|
||||
|
||||
// RandomAll 按权重随机选取 N 个不重复的元素
|
||||
func (m *randMod) RandomAll(data map[int]int, n int) []int {
|
||||
if n > len(data) {
|
||||
return nil
|
||||
}
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
// 复制权重映射,避免修改原始数据
|
||||
remainingWeights := make(map[int]int)
|
||||
for k, v := range data {
|
||||
remainingWeights[k] = v
|
||||
}
|
||||
result := make([]int, 0, n)
|
||||
|
||||
for i := 0; i < n; i++ {
|
||||
totalWeight := 0
|
||||
// 计算剩余元素的总权重
|
||||
for _, weight := range remainingWeights {
|
||||
totalWeight += weight
|
||||
}
|
||||
if totalWeight == 0 {
|
||||
break
|
||||
}
|
||||
// 生成一个 0 到总权重之间的随机数
|
||||
randomNum := rand.Intn(totalWeight)
|
||||
currentWeight := 0
|
||||
for key, weight := range remainingWeights {
|
||||
currentWeight += weight
|
||||
if randomNum < currentWeight {
|
||||
// 将选中的元素添加到结果切片中
|
||||
result = append(result, key)
|
||||
// 从剩余权重映射中移除选中的元素
|
||||
delete(remainingWeights, key)
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
@@ -124,11 +124,11 @@ func (m *tools) Items2Map(items [][]int64) (list map[int64]int64) {
|
||||
// 该函数通过遍历切片,从后向前检查每个元素,如果找到与指定值相等的元素,则将其从切片中移除。
|
||||
// 这种从后向前的遍历方法可以避免因移除元素而导致的数组重新排列带来的额外计算。
|
||||
// RemoveSlice 删除切片中的某个值
|
||||
func RemoveSlice[t Number](slice []t, value t) []t {
|
||||
func RemoveSlice[t Number](slice []t, value ...t) []t {
|
||||
// 从后向前遍历切片
|
||||
for i := len(slice) - 1; i >= 0; i-- {
|
||||
// 检查当前元素是否等于需要移除的值
|
||||
if slice[i] == value {
|
||||
if InArray(slice[i], value) {
|
||||
// 如果相等,移除该元素
|
||||
// 使用append和切片操作符来实现移除操作,将i之前和i之后的元素合并到一起
|
||||
slice = append(slice[:i], slice[i+1:]...)
|
||||
|
||||
Reference in New Issue
Block a user