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

math.combinatorics: use nth-unsafe more.

db4
John Benediktsson 2013-04-06 12:43:02 -07:00
parent d15dbaebb4
commit 64111e1342
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -49,10 +49,13 @@ MEMO: factorial ( n -- n! )
: permutation-iota ( seq -- iota ) : permutation-iota ( seq -- iota )
length factorial iota ; inline length factorial iota ; inline
: nths-unsafe ( indices seq -- seq' )
[ [ nth-unsafe ] curry ] keep map-as ;
PRIVATE> PRIVATE>
: permutation ( n seq -- seq' ) : permutation ( n seq -- seq' )
[ permutation-indices ] keep nths ; [ permutation-indices ] keep nths-unsafe ;
TUPLE: permutations length seq ; TUPLE: permutations length seq ;
@ -139,13 +142,13 @@ PRIVATE>
:: combinations-quot ( seq k quot -- seq quot ) :: combinations-quot ( seq k quot -- seq quot )
seq length :> n seq length :> n
n k nCk iota [ n k nCk iota [
k n combination-indices seq nths quot call k n combination-indices seq nths-unsafe quot call
] ; inline ] ; inline
PRIVATE> PRIVATE>
: combination ( m seq k -- seq' ) : combination ( m seq k -- seq' )
swap [ length combination-indices ] [ nths ] bi ; swap [ length combination-indices ] [ nths-unsafe ] bi ;
TUPLE: combinations seq k length ; TUPLE: combinations seq k length ;