exercism/go/prime-factors/prime.go

50 lines
963 B
Go
Raw Normal View History

2016-08-23 20:47:19 +00:00
package prime
2016-08-30 01:53:59 +00:00
import "fmt"
2016-08-25 17:17:37 +00:00
2016-08-23 20:47:19 +00:00
const testVersion = 2
2016-08-25 17:17:37 +00:00
var sieve []int64
2016-08-30 01:53:59 +00:00
var iter int
2016-08-25 17:17:37 +00:00
func Factors(num int64) []int64 {
2016-08-30 01:53:59 +00:00
fmt.Println("Factors Called on ", num)
if num == 1 {
return []int64{}
}
2016-08-25 17:17:37 +00:00
for i := range sieve {
if sieve[i] == num {
2016-08-30 01:53:59 +00:00
// It's a prime already in the sieve
2016-08-25 17:17:37 +00:00
return []int64{num}
}
2016-08-30 01:53:59 +00:00
if num%sieve[i] == 0 {
div := (num / sieve[i])
return append([]int64{sieve[i]}, Factors(div)...)
}
2016-08-25 17:17:37 +00:00
}
2016-08-30 01:53:59 +00:00
// If we're here, we don't have the prime factors for this number
// Add 1 to the highest prime that we have and find the next prime
i := int64(2)
if len(sieve) > 0 {
i = sieve[len(sieve)-1]
}
fmt.Println(" Looking for a new prime starting at ", i)
for ; i <= num; i++ {
isPrime := true
for j := range sieve {
if i%sieve[j] == 0 {
isPrime = false
}
if !isPrime {
break
}
}
if isPrime {
sieve = append(sieve, i)
// New prime in the sieve, try again
return Factors(num)
2016-08-23 20:47:19 +00:00
}
}
2016-08-30 01:53:59 +00:00
return []int64{}
2016-08-23 20:47:19 +00:00
}