142 lines
2.6 KiB
Go
142 lines
2.6 KiB
Go
package maths
|
||
|
||
import (
|
||
"github.com/kercylan98/minotaur/utils/generic"
|
||
"math"
|
||
"sort"
|
||
)
|
||
|
||
const (
|
||
DefaultTolerance = 0.0001 // 默认误差范围
|
||
Zero = 0 // 零
|
||
)
|
||
|
||
// GetDefaultTolerance 获取默认误差范围
|
||
func GetDefaultTolerance() float64 {
|
||
return DefaultTolerance
|
||
}
|
||
|
||
// Pow 整数幂运算
|
||
func Pow(a, n int) int {
|
||
if a == 0 {
|
||
return 0
|
||
}
|
||
if n == 0 {
|
||
return 1
|
||
}
|
||
if n == 1 {
|
||
return a
|
||
}
|
||
var result int = 1
|
||
factor := a
|
||
for n != 0 {
|
||
if n&1 != 0 {
|
||
// 当前位是 1，需要乘进去
|
||
result *= factor
|
||
}
|
||
factor *= factor
|
||
n = n >> 1
|
||
}
|
||
return result
|
||
}
|
||
|
||
// PowInt64 整数幂运算
|
||
func PowInt64(a, n int64) int64 {
|
||
if a == 0 {
|
||
return 0
|
||
}
|
||
if n == 0 {
|
||
return 1
|
||
}
|
||
if n == 1 {
|
||
return a
|
||
}
|
||
var result int64 = 1
|
||
factor := a
|
||
for n != 0 {
|
||
if n&1 != 0 {
|
||
// 当前位是 1，需要乘进去
|
||
result *= factor
|
||
}
|
||
factor *= factor
|
||
n = n >> 1
|
||
}
|
||
return result
|
||
}
|
||
|
||
// Min 返回两个数之中较小的值
|
||
func Min[V generic.Number](a, b V) V {
|
||
if a < b {
|
||
return a
|
||
}
|
||
return b
|
||
}
|
||
|
||
// Max 返回两个数之中较大的值
|
||
func Max[V generic.Number](a, b V) V {
|
||
if a > b {
|
||
return a
|
||
}
|
||
return b
|
||
}
|
||
|
||
// MinMax 将两个数按照较小的和较大的顺序进行返回
|
||
func MinMax[V generic.Number](a, b V) (min, max V) {
|
||
if a < b {
|
||
return a, b
|
||
}
|
||
return b, a
|
||
}
|
||
|
||
// MaxMin 将两个数按照较大的和较小的顺序进行返回
|
||
func MaxMin[V generic.Number](a, b V) (max, min V) {
|
||
if a > b {
|
||
return a, b
|
||
}
|
||
return b, a
|
||
}
|
||
|
||
// Clamp 将给定值限制在最小值和最大值之间
|
||
func Clamp[V generic.Number](value, min, max V) V {
|
||
if value < min {
|
||
return min
|
||
}
|
||
if value > max {
|
||
return max
|
||
}
|
||
return value
|
||
}
|
||
|
||
// Tolerance 检查两个值是否在一个误差范围内
|
||
func Tolerance[V generic.Number](value1, value2, tolerance V) bool {
|
||
return V(math.Abs(float64(value1-value2))) <= tolerance
|
||
}
|
||
|
||
// Merge 通过一个参考值合并两个数字
|
||
func Merge[V generic.SignedNumber](refer, a, b V) V {
|
||
return b*refer + a
|
||
}
|
||
|
||
// UnMerge 通过一个参考值取消合并的两个数字
|
||
func UnMerge[V generic.SignedNumber](refer, num V) (a, b V) {
|
||
a = V(math.Mod(float64(num), float64(refer)))
|
||
b = num / refer
|
||
return a, b
|
||
}
|
||
|
||
// ToContinuous 将一组非连续的数字转换为从1开始的连续数字
|
||
// - 返回值是一个 map，key 是从 1 开始的连续数字，value 是原始数字
|
||
func ToContinuous[V generic.Integer](nums []V) map[V]V {
|
||
if len(nums) == 0 {
|
||
return nil
|
||
}
|
||
sort.Slice(nums, func(i, j int) bool {
|
||
return nums[i] < nums[j]
|
||
})
|
||
var result = make(map[V]V)
|
||
for i, num := range nums {
|
||
result[V(i+1)] = num
|
||
}
|
||
return result
|
||
}
|