118 lines
2.1 KiB
Go
118 lines
2.1 KiB
Go
// Package number
|
|
// 使用随机数时需要先 调用 rand.seed()函数
|
|
package number
|
|
|
|
import (
|
|
"fmt"
|
|
"golang.org/x/exp/constraints"
|
|
"math"
|
|
"math/rand"
|
|
"strconv"
|
|
)
|
|
|
|
func Range[T constraints.Integer](start, end T, steps ...T) []T {
|
|
step := T(1)
|
|
if len(steps) > 0 {
|
|
step = steps[0]
|
|
}
|
|
var l int
|
|
if step == 0 {
|
|
l = int(end - start + 1)
|
|
} else {
|
|
l = int((end-start+1)/step + 1)
|
|
}
|
|
if l < 0 {
|
|
l = -l
|
|
}
|
|
r := make([]T, 0, l)
|
|
gap := start
|
|
for i := 0; i < l; i++ {
|
|
r = append(r, gap)
|
|
gap += step
|
|
}
|
|
return r
|
|
}
|
|
|
|
// Rand 都为闭区间 [start,end]
|
|
func Rand[T constraints.Integer](start, end T) T {
|
|
end++
|
|
return T(rand.Int63n(int64(end-start))) + start
|
|
}
|
|
|
|
func Min[T constraints.Integer | constraints.Float](a ...T) T {
|
|
mins := a[0]
|
|
for _, t := range a {
|
|
if mins > t {
|
|
mins = t
|
|
}
|
|
}
|
|
return mins
|
|
}
|
|
|
|
func Max[T constraints.Integer | constraints.Float](a ...T) T {
|
|
maxs := a[0]
|
|
for _, t := range a {
|
|
if maxs < t {
|
|
maxs = t
|
|
}
|
|
}
|
|
return maxs
|
|
}
|
|
|
|
func Sum[T constraints.Integer | constraints.Float](a ...T) T {
|
|
s := T(0)
|
|
for _, t := range a {
|
|
s += t
|
|
}
|
|
return s
|
|
}
|
|
|
|
func Add[T constraints.Integer | constraints.Float](i, j T) T {
|
|
return i + j
|
|
}
|
|
func Sub[T constraints.Integer | constraints.Float](i, j T) T {
|
|
return i - j
|
|
}
|
|
|
|
func ToString[T constraints.Integer | constraints.Float](n T) string {
|
|
return fmt.Sprintf("%v", n)
|
|
}
|
|
|
|
func IntToString[T constraints.Integer](i T) string {
|
|
return strconv.FormatInt(int64(i), 10)
|
|
}
|
|
|
|
func Abs[T constraints.Integer | constraints.Float](n T) T {
|
|
if n >= 0 {
|
|
return n
|
|
}
|
|
return -n
|
|
}
|
|
|
|
func Mul[T constraints.Integer | constraints.Float](i, j T) T {
|
|
return i * j
|
|
}
|
|
|
|
func Divide[T constraints.Integer | constraints.Float](i, j T) T {
|
|
return i / j
|
|
}
|
|
|
|
func DivideCeil[T constraints.Integer](num1, num2 T) T {
|
|
return T(math.Ceil(float64(num1) / float64(num2)))
|
|
}
|
|
|
|
type Counter[T constraints.Integer] func() T
|
|
|
|
func Counters[T constraints.Integer]() func() T {
|
|
var count T
|
|
return func() T {
|
|
count++
|
|
return count
|
|
}
|
|
}
|
|
|
|
func Round(f float64, precision int) float64 {
|
|
p := math.Pow10(precision)
|
|
return math.Floor(f*p+0.5) / p
|
|
}
|