summaryrefslogtreecommitdiff
path: root/factors/prime_factorization.go
diff options
context:
space:
mode:
authorAdrien Hopkins <adrien.p.hopkins@gmail.com>2023-08-07 15:24:54 -0500
committerAdrien Hopkins <adrien.p.hopkins@gmail.com>2023-08-18 14:26:13 -0500
commit8c6419bde0cb6893cf7e789c8fceaea5638c95ff (patch)
tree234fe0c4249d68bbb402b5bfd4f3d58fea832dd0 /factors/prime_factorization.go
parent1bef5152dfa454b308946791ad7a8efc747e3fbd (diff)
Add tests for Prime Factorization function
Diffstat (limited to 'factors/prime_factorization.go')
-rw-r--r--factors/prime_factorization.go4
1 files changed, 4 insertions, 0 deletions
diff --git a/factors/prime_factorization.go b/factors/prime_factorization.go
index 62498b5..3f3abc7 100644
--- a/factors/prime_factorization.go
+++ b/factors/prime_factorization.go
@@ -15,6 +15,10 @@ type PrimeFactorization struct {
// PrimeFactorize creates a PrimeFactorization by factorizing a number.
func PrimeFactorize(n uint) PrimeFactorization {
+ if n == 0 {
+ return PrimeFactorization{map[uint]uint{0: 1}}
+ }
+
unfactored := n // number with all found factors removed (divided)
exponents := make(map[uint]uint)