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

View File

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