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

clean up math.combinatorics and add another combinator

db4
Doug Coleman 2009-11-27 22:50:06 -06:00
parent cdc1f1e041
commit a1bb120498
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
basis/math/combinatorics/combinatorics.factor
View File

@ -103,23 +103,26 @@ C: <combo> combo
: apply-combination ( m combo -- seq ) : apply-combination ( m combo -- seq )
[ combination-indices ] keep seq>> nths ; [ combination-indices ] keep seq>> nths ;
: combinations-quot ( seq k quot -- seq quot )
[ <combo> [ choose [0,b) ] keep ] dip
'[ _ apply-combination @ ] ; inline
PRIVATE> PRIVATE>
: each-combination ( seq k quot -- )
combinations-quot each ; inline
: map-combinations ( seq k quot -- )
combinations-quot map ; inline
: map>assoc-combinations ( seq k quot exemplar -- )
[ combinations-quot ] dip map>assoc ; inline
: combination ( m seq k -- seq ) : combination ( m seq k -- seq )
<combo> apply-combination ; <combo> apply-combination ;
: all-combinations ( seq k -- seq ) : all-combinations ( seq k -- seq )
<combo> [ choose [0,b) ] keep [ ] combinations-quot map ;
'[ _ apply-combination ] map ;
: each-combination ( seq k quot -- )
[ <combo> [ choose [0,b) ] keep ] dip
'[ _ apply-combination @ ] each ; inline
: map-combinations ( seq k quot -- )
[ <combo> [ choose [0,b) ] keep ] dip
'[ _ apply-combination @ ] map ; inline
: reduce-combinations ( seq k identity quot -- result ) : reduce-combinations ( seq k identity quot -- result )
[ -rot ] dip each-combination ; inline [ -rot ] dip each-combination ; inline