(use srfi.13) ; string (use srfi.42) ; list-ec (use srfi.197) ; chain (use gauche.array) (use gauche.collection) (use gauche.dictionary) (use gauche.generator) (use gauche.sequence) (use scheme.ideque) (use scheme.list) (use scheme.set) (use scheme.sort) (use util.combinations) (use util.match) (use data.queue) (define (input) (string-trim-both (read-line))) (define (ii) (string->number (input))) (define (li) (let ((s (input))) (map string->number (string-split s " ")))) (define (prn . args) (prn* args)) (define-method prn* ((seq )) (for-each-with-index (^(i x) (when (> i 0) (display " ")) (display x)) seq) (newline)) (define (prn-yn b) (prn (if b "Yes" "No"))) (define-syntax ->/prn (syntax-rules () ((_ x fns ...) (-> x fns ... prn)))) (define-syntax prn/d (syntax-rules () ((_ expr ...) (with-output-to-port (current-error-port) (^() (prn expr ...)))))) (define int string->number) (define str x->string) (define string* (apply$ string)) (define-method min ((xs )) (fold min (~ xs 0) xs)) (define-method max ((xs )) (fold max (~ xs 0) xs)) (define (minmax . xs) (values->list (apply min&max xs))) (define-method minmax ((xs )) (values->list (apply min&max xs))) (define (sum xs) (fold + 0 xs)) (define (divmod a b) (values->list (div-and-mod a b))) (define (ceildiv a b) (div (+ a (1- b)) b)) (define (1+ n) (+ n 1)) (define (1- n) (- n 1)) (define (!= a b) (not (= a b))) (define (midpoint a b) (div (+ a b) 2)) (define pow (case-lambda ((a b) (expt a b)) ((a b m) (expt-mod a b m)))) (define gcd* (apply$ gcd)) (define isqrt exact-integer-sqrt) (define ++ string-append) (define zip (map$ list)) (define all every) (define any-ec any?-ec) (define all-ec every?-ec) (define concat concatenate) (define (pairwise xs) (zip xs (cdr xs))) (define (comb n k) (if (or (< k 0) (> k n)) 0 (let loop ((i 0) (x 1)) (if (= i k) x (loop (1+ i) (div (* x (- n i)) (1+ i))))))) #; (define-method frequencies ((xs )) (rlet1 dict (make-dict) (for-each (^x (dict-update! dict x 1+ 0)) xs))) (define-method frequencies ((xs )) (let1 ht (make-dict) (for-each (^x (dict-update! ht x 1+ 0)) xs) (match-lambda* (('get key) (hash-table-get ht key 0)) (('get key default) (hash-table-get ht key default)) (('keys) (hash-table-keys ht)) (('values) (hash-table-values ht)) (('contains? key) (hash-table-contains? ht key)) (('items) (hash-table->alist ht))))) (define-macro (input! vars . body) (let ((binds (map (^x (cond ((symbol? x) `((,x) (values (ii)))) ((list? x) (if (= 1 (length x)) `(,x (values (li))) `(,x (apply values (li))))) (else (error "symbol or list required, but got:" x)))) vars))) `(let*-values ,binds ,@body))) (define-macro (! self quoted-name . args) (let ((name (cadr quoted-name))) `(,name ,self ,@args))) (define-macro (d/ . args) (let ((ss (map (^(expr) `(list ,(x->string expr) ,expr)) args))) `(prn/d (list ,@ss)))) (define mlet match-let) (define mlet* match-let*) (define mlet1 match-let1) (define-macro (mfn pat . body) (let ((arg (gensym))) `(lambda (,arg) (mlet1 ,pat ,arg ,@body)))) (define-syntax count-ec (syntax-rules () ((_ qualifiers ...) (sum-ec qualifiers ... 1)))) (define-method len ((coll )) (size-of coll)) (define-method len ((deq )) (ideque-length deq)) (define (accum xs) (define (proc a b) (let1 t (+ a b) (values t t))) (map-accum proc 0 xs)) (define (digits n) (map digit->integer (str n))) (define (digits->int ds) (fold-left (^(a b) (+ (* 10 a) b)) 0 ds)) (define (-> x . fns) (call-with-values (^() (values x)) (apply compose (reverse fns)))) (define (rep n thunk) (list-ec (: _ n) (thunk))) (define (memoize fn) (let ((cache (make-hash-table 'equal?))) (lambda args (if (hash-table-exists? cache args) (hash-table-get cache args) (let ((val (apply fn args))) (hash-table-put! cache args val) val))))) (define (zip-longest . args) (let* ((n (apply max (map length args))) (xxs (map (^(xs) (append xs (make-list (- n (length xs)) #f))) args))) (map (pa$ delete #f) (apply zip xxs)))) (define (set-from xs) (apply set eqv-comparator xs)) (define (difference xs ys) (let ((excludes (set-from ys))) (filter (^x (not (set-contains? excludes x))) xs))) ;; (define (make-dict) (make-hash-table eqv-comparator)) (define-method get ((ht ) key :optional (default 0)) (hash-table-get ht key default)) (define-method put! ((ht ) key value) (hash-table-put! ht key value)) (define-method keys ((ht )) (hash-table-keys ht)) (define-method values ((ht )) (hash-table-values ht)) ;; (define-method back ((deq )) (ideque-back deq)) (define-method append ((deq ) x) (ideque-add-back deq x)) (define-method pop ((deq )) (ideque-remove-back deq)) (use math.prime) (define (f x) (let ((freq (frequencies (naive-factorize x)))) freq)) (input! ((X A Y B)) (let ((xps (f X)) (yps (f Y))) (-> (all (mfn (k . v) (let ((s (* (xps'get k) A)) (t (* v B))) (>= s t))) (yps'items)) prn-yn)))