Fannkuch benchmark from shootout

db4
Slava Pestov 2008-11-30 09:06:42 -06:00
parent c7c3bbc5f5
commit b4e47eb1e1
1 changed files with 35 additions and 0 deletions

View File

@ -0,0 +1,35 @@
! Copyright (C) 2008 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
USING: kernel fry math math.combinatorics math.order sequences
io prettyprint ;
IN: benchmark.fannkuch
: count ( quot: ( -- ? ) -- n )
#! Call quot until it returns false, return number of times
#! it was true
[ 0 ] dip '[ _ dip swap [ [ 1+ ] when ] keep ] loop ; inline
: count-flips ( perm -- flip# )
'[
_ dup first dup 1 =
[ 2drop f ] [ head-slice reverse-here t ] if
] count ; inline
: write-permutation ( perm -- )
[ CHAR: 0 + write1 ] each nl ; inline
: fannkuch-step ( counter max-flips perm -- counter max-flips )
pick 30 < [ [ 1+ ] [ ] [ dup write-permutation ] tri* ] when
count-flips max ; inline
: fannkuch ( n -- flip# )
[
[ 0 0 ] dip [ 1+ ] B{ } map-as
[ fannkuch-step ] each-permutation nip
] keep
"Pfannkuchen(" write pprint ") = " write . ;
: fannkuch-main ( -- )
9 fannkuch ;
MAIN: fannkuch-main