sequences: Add some short useful factors. Clean up the use of longest.

db4
Doug Coleman 2013-03-11 17:41:21 -07:00
parent ce72121c7a
commit 875e45a640
9 changed files with 34 additions and 39 deletions

View File

@ -46,12 +46,9 @@ SYMBOLS: return-addresses gc-maps ;
compiled-offset return-addresses get push compiled-offset return-addresses get push
] [ drop ] if ; ] [ drop ] if ;
: longest ( seqs -- n )
[ length ] [ max ] map-reduce ;
: emit-scrub ( seqs -- n ) : emit-scrub ( seqs -- n )
! seqs is a sequence of sequences of 0/1 ! seqs is a sequence of sequences of 0/1
dup longest dup longest length
[ '[ [ 0 = ] ?{ } map-as _ f pad-tail % ] each ] keep ; [ '[ [ 0 = ] ?{ } map-as _ f pad-tail % ] each ] keep ;
: integers>bits ( seq n -- bit-array ) : integers>bits ( seq n -- bit-array )

View File

@ -21,7 +21,7 @@ SYMBOLS: +bottom+ +top+ ;
! Terminated branches are padded with bottom values which ! Terminated branches are padded with bottom values which
! unify with literals. ! unify with literals.
dup empty? [ dup empty? [
dup [ length ] [ max ] map-reduce dup longest length
'[ _ +bottom+ pad-head ] map '[ _ +bottom+ pad-head ] map
] unless ; ] unless ;

View File

@ -5,14 +5,11 @@ IN: strings.tables
<PRIVATE <PRIVATE
: max-length ( seq -- n )
[ length ] [ max ] map-reduce ; inline
: format-row ( seq -- seq ) : format-row ( seq -- seq )
dup max-length '[ _ "" pad-tail ] map! ; dup longest length '[ _ "" pad-tail ] map! ;
: format-column ( seq -- seq ) : format-column ( seq -- seq )
dup max-length '[ _ CHAR: \s pad-tail ] map! ; dup longest length '[ _ CHAR: \s pad-tail ] map! ;
PRIVATE> PRIVATE>

View File

@ -844,6 +844,9 @@ PRIVATE>
[ append ] padding ; [ append ] padding ;
: shorter? ( seq1 seq2 -- ? ) [ length ] bi@ < ; : shorter? ( seq1 seq2 -- ? ) [ length ] bi@ < ;
: longer? ( seq1 seq2 -- ? ) [ length ] bi@ > ;
: shorter ( seq1 seq2 -- seq ) [ [ length ] bi@ <= ] 2keep ? ; inline
: longer ( seq1 seq2 -- seq ) [ [ length ] bi@ >= ] 2keep ? ; inline
: head? ( seq begin -- ? ) : head? ( seq begin -- ? )
2dup shorter? [ 2dup shorter? [
@ -1013,6 +1016,16 @@ M: object sum 0 [ + ] binary-reduce ; inline
: cartesian-product ( seq1 seq2 -- newseq ) : cartesian-product ( seq1 seq2 -- newseq )
[ { } 2sequence ] cartesian-map ; [ { } 2sequence ] cartesian-map ;
: filter-length ( seq n -- seq' ) [ swap length = ] curry filter ;
: shortest ( seqs -- elt ) [ ] [ shorter ] map-reduce ;
: longest ( seqs -- elt ) [ ] [ longer ] map-reduce ;
: all-shortest ( seqs -- seqs' ) dup shortest length filter-length ;
: all-longest ( seqs -- seqs' ) dup longest length filter-length ;
! We hand-optimize flip to such a degree because type hints ! We hand-optimize flip to such a degree because type hints
! cannot express that an array is an array of arrays yet, and ! cannot express that an array is an array of arrays yet, and
! this word happens to be performance-critical since the compiler ! this word happens to be performance-critical since the compiler

View File

@ -2,34 +2,27 @@
! See http://factorcode.org/license.txt for BSD license ! See http://factorcode.org/license.txt for BSD license
USING: arrays ascii assocs fry io.encodings.ascii io.files USING: arrays ascii assocs fry io.encodings.ascii io.files
kernel math math.order memoize sequences sorting ; kernel math math.order memoize sequences sorting
math.statistics ;
FROM: sets => members ;
IN: anagrams IN: anagrams
: (all-anagrams) ( seq assoc -- ) : make-anagram-hash ( strings -- assoc )
'[ dup natural-sort _ push-at ] each ; [ natural-sort ] collect-by
[ members ] assoc-map
: all-anagrams ( seq -- assoc )
H{ } clone [ (all-anagrams) ] keep
[ nip length 1 > ] assoc-filter ; [ nip length 1 > ] assoc-filter ;
MEMO: dict-words ( -- seq ) MEMO: dict-words ( -- seq )
"/usr/share/dict/words" ascii file-lines [ >lower ] map ; "/usr/share/dict/words" ascii file-lines [ >lower ] map ;
MEMO: dict-anagrams ( -- assoc ) MEMO: dict-anagrams ( -- assoc )
dict-words all-anagrams ; dict-words make-anagram-hash ;
: anagrams ( str -- seq/f ) : anagrams ( str -- seq/f )
>lower natural-sort dict-anagrams at ; >lower natural-sort dict-anagrams at ;
: longest ( seq -- subseq )
dup 0 [ length max ] reduce '[ length _ = ] filter ;
: most-anagrams ( -- seq ) : most-anagrams ( -- seq )
dict-anagrams values longest ; dict-anagrams values all-longest ;
: longest-anagrams ( -- seq ) : longest-anagrams ( -- seq )
dict-anagrams [ keys longest ] keep '[ _ at ] map ; dict-anagrams [ keys all-longest ] keep '[ _ at ] map ;

View File

@ -38,16 +38,13 @@ IN: project-euler.014
: next-collatz ( n -- n ) : next-collatz ( n -- n )
dup even? [ 2 / ] [ 3 * 1 + ] if ; dup even? [ 2 / ] [ 3 * 1 + ] if ;
: longest ( seq seq -- seq )
2dup [ length ] bi@ > [ drop ] [ nip ] if ;
PRIVATE> PRIVATE>
: collatz ( n -- seq ) : collatz ( n -- seq )
[ [ dup 1 > ] [ dup , next-collatz ] while , ] { } make ; [ [ dup 1 > ] [ dup , next-collatz ] while , ] { } make ;
: euler014 ( -- answer ) : euler014 ( -- answer )
1000000 [1,b] { } [ collatz longest ] reduce first ; 1000000 [1,b] { } [ collatz longer ] reduce first ;
! [ euler014 ] time ! [ euler014 ] time
! 52868 ms run / 483 ms GC time ! 52868 ms run / 483 ms GC time
@ -65,7 +62,7 @@ PRIVATE>
: euler014a ( -- answer ) : euler014a ( -- answer )
500000 1000000 [a,b] { 1 } [ 500000 1000000 [a,b] { 1 } [
dup worth-calculating? [ collatz longest ] [ drop ] if dup worth-calculating? [ collatz longer ] [ drop ] if
] reduce first ; ] reduce first ;
! [ euler014a ] 10 ave-time ! [ euler014a ] 10 ave-time

View File

@ -1,6 +1,7 @@
! Copyright (c) 2008 Aaron Schaefer. ! Copyright (c) 2008 Aaron Schaefer.
! See http://factorcode.org/license.txt for BSD license. ! See http://factorcode.org/license.txt for BSD license.
USING: arrays kernel locals math math.primes sequences project-euler.common ; USING: arrays kernel locals math math.order math.primes
project-euler.common sequences ;
IN: project-euler.050 IN: project-euler.050
! http://projecteuler.net/index.php?section=problems&id=50 ! http://projecteuler.net/index.php?section=problems&id=50
@ -62,14 +63,11 @@ IN: project-euler.050
[ length ] dip 2array [ length ] dip 2array
] if ; ] if ;
: longest ( pair pair -- longest )
2dup [ first ] bi@ > [ drop ] [ nip ] if ;
: continue? ( pair seq -- ? ) : continue? ( pair seq -- ? )
[ first ] [ length 1 - ] bi* < ; [ first ] [ length 1 - ] bi* < ;
: (find-longest) ( best seq limit -- best ) : (find-longest) ( best seq limit -- best )
[ longest-prime longest ] 2keep 2over continue? [ [ longest-prime max ] 2keep 2over continue? [
[ rest-slice ] dip (find-longest) [ rest-slice ] dip (find-longest)
] [ 2drop ] if ; ] [ 2drop ] if ;

View File

@ -45,11 +45,11 @@ justified,$right$justified,$or$center$justified$within$its$column."
: split-and-pad ( text -- lines ) : split-and-pad ( text -- lines )
"\n" split [ "$" split harvest ] map "\n" split [ "$" split harvest ] map
dup [ length ] [ max ] map-reduce dup longest length
'[ _ "" pad-tail ] map ; '[ _ "" pad-tail ] map ;
: column-widths ( columns -- widths ) : column-widths ( columns -- widths )
[ [ length ] [ max ] map-reduce ] map ; [ longest length ] map ;
SINGLETONS: +left+ +middle+ +right+ ; SINGLETONS: +left+ +middle+ +right+ ;

View File

@ -217,7 +217,7 @@ PRIVATE>
: round-robin ( seq -- newseq ) : round-robin ( seq -- newseq )
[ { } ] [ [ { } ] [
[ [ length ] [ max ] map-reduce iota ] keep [ longest length iota ] keep
[ [ ?nth ] with map ] curry map concat sift [ [ ?nth ] with map ] curry map concat sift
] if-empty ; ] if-empty ;