avoid stack overflow if k > n

2025-09-07 23:07:30 -04:00
parent 2d359b3e63
commit 93ba12400f
1 changed files with 2 additions and 0 deletions
--- a/cmd/partitions.go
+++ b/cmd/partitions.go
@@ -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)