steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Short circuit trivial composites

db4
Samuel Tardieu 2009-06-24 15:36:45 +02:00
parent 50191588e4
commit 094c82c50b
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
basis/math/primes/primes.factor
View File

@ -1,9 +1,9 @@
! Copyright (C) 2007-2009 Samuel Tardieu. ! Copyright (C) 2007-2009 Samuel Tardieu.
! See http://factorcode.org/license.txt for BSD license. ! See http://factorcode.org/license.txt for BSD license.
USING: combinators fry kernel math math.bitwise math.functions USING: combinators combinators.short-circuit fry kernel math
math.order math.primes.erato math.primes.erato.private math.bitwise math.functions math.order math.primes.erato
math.primes.miller-rabin math.ranges literals random sequences sets math.primes.erato.private math.primes.miller-rabin math.ranges
vectors ; literals random sequences sets vectors ;
IN: math.primes IN: math.primes
<PRIVATE <PRIVATE
@ -28,12 +28,14 @@ IN: math.primes
108 max 10000 min <vector> ] keep 108 max 10000 min <vector> ] keep
3 < [ [ 2 swap push ] keep ] when ; 3 < [ [ 2 swap push ] keep ] when ;
: simple? ( n -- ? ) { [ even? ] [ 3 mod 0 = ] [ 5 mod 0 = ] } 1|| ;
PRIVATE> PRIVATE>
: prime? ( n -- ? ) : prime? ( n -- ? )
{ {
{ [ dup 7 < ] [ { 2 3 5 } member? ] } { [ dup 7 < ] [ { 2 3 5 } member? ] }
{ [ dup even? ] [ 2 = ] } { [ dup simple? ] [ drop f ] }
[ (prime?) ] [ (prime?) ]
} cond ; foldable } cond ; foldable