Update Pow

itsubaki · itsubaki · commit 4de42405501d · 2026-03-15T19:44:23.000+09:00
diff --git a/math/number/pow.go b/math/number/pow.go
@@ -8,7 +8,7 @@ func Ldexp(a, b int) float64 {
 }
 
 // Pow returns a**r, the base-a exponential of r.
-func Pow(a, r int) int {
+func Pow[T ~int | ~int64](a, r T) T {
 	if a == 0 {
 		return 0
 	}
@@ -18,7 +18,7 @@ func Pow(a, r int) int {
 	}
 
 	// exponentiation by squaring
-	p := 1
+	p := T(1)
 	for r > 0 {
 		if r&1 == 1 {
 			p = p * a

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@ func Ldexp(a, b int) float64 {`
`8`	`8`	`}`
`9`	`9`
`10`	`10`	`// Pow returns a**r, the base-a exponential of r.`
`11`		`-func Pow(a, r int) int {`
	`11`	`+func Pow[T ~int \| ~int64](a, r T) T {`
`12`	`12`	`if a == 0 {`
`13`	`13`	`return 0`
`14`	`14`	`}`
`@@ -18,7 +18,7 @@ func Pow(a, r int) int {`
`18`	`18`	`}`
`19`	`19`
`20`	`20`	`// exponentiation by squaring`
`21`		`- p := 1`
	`21`	`+ p := T(1)`
`22`	`22`	`for r > 0 {`
`23`	`23`	`if r&1 == 1 {`
`24`	`24`	`p = p * a`