filifa
/
mathtools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

avoid stack overflow if k > n

This commit is contained in:
filifa 2025-09-07 23:07:30 -04:00
parent 2d359b3e63
commit 93ba12400f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/partitions.go
View File

@ -34,6 +34,8 @@ func p(n, k *big.Int) *big.Int {
return big.NewInt(1) return big.NewInt(1)
} else if k.Cmp(big.NewInt(0)) == 0 { } else if k.Cmp(big.NewInt(0)) == 0 {
return big.NewInt(0) return big.NewInt(0)
} else if k.Cmp(n) == 1 {
return p(n, n)
} }
newN := new(big.Int).Sub(n, k) newN := new(big.Int).Sub(n, k)