Compare commits

...

3 Commits

Author SHA1 Message Date
ayflying
4963780a32 修改删除结构体方法,支持传递多个数据进行删除 2025-03-13 16:25:28 +08:00
ayflying
70ebaae3d2 增加权重随机 2025-03-13 15:33:13 +08:00
ayflying
9cd286fd92 增加权重随机方法 2025-03-13 14:28:26 +08:00
2 changed files with 54 additions and 2 deletions

52
tools/random.go Normal file
View 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
}

View File

@@ -124,11 +124,11 @@ func (m *tools) Items2Map(items [][]int64) (list map[int64]int64) {
// 该函数通过遍历切片,从后向前检查每个元素,如果找到与指定值相等的元素,则将其从切片中移除。 // 该函数通过遍历切片,从后向前检查每个元素,如果找到与指定值相等的元素,则将其从切片中移除。
// 这种从后向前的遍历方法可以避免因移除元素而导致的数组重新排列带来的额外计算。 // 这种从后向前的遍历方法可以避免因移除元素而导致的数组重新排列带来的额外计算。
// RemoveSlice 删除切片中的某个值 // 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-- { for i := len(slice) - 1; i >= 0; i-- {
// 检查当前元素是否等于需要移除的值 // 检查当前元素是否等于需要移除的值
if slice[i] == value { if InArray(slice[i], value) {
// 如果相等,移除该元素 // 如果相等,移除该元素
// 使用append和切片操作符来实现移除操作将i之前和i之后的元素合并到一起 // 使用append和切片操作符来实现移除操作将i之前和i之后的元素合并到一起
slice = append(slice[:i], slice[i+1:]...) slice = append(slice[:i], slice[i+1:]...)