結果
| 問題 |
No.2865 Base 10 Subsets 2
|
| コンテスト | |
| ユーザー |
 norioc
|
| 提出日時 | 2025-07-29 20:00:56 |
| 言語 | Scheme (Gauche-0.9.15) |
| 結果 |
AC
|
| 実行時間 | 140 ms / 2,000 ms |
| コード長 | 6,545 bytes |
| コンパイル時間 | 115 ms |
| コンパイル使用メモリ | 7,716 KB |
| 実行使用メモリ | 30,208 KB |
| 最終ジャッジ日時 | 2025-07-29 20:01:01 |
| 合計ジャッジ時間 | 4,132 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge2 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 12 |
ソースコード
(use srfi.13) ; string
(use srfi.42) ; list-ec
(use srfi.197) ; chain
(use data.queue)
(use gauche.collection)
(use gauche.dictionary)
(use gauche.generator)
(use gauche.sequence)
(use scheme.ideque)
(use scheme.list)
(use scheme.vector)
(use util.match)
(define input read-line)
(define (ii)
(string->number (input)))
(define (li)
(let ((s (input)))
(map string->number (string-split s " "))))
(define-method prn* ((seq <sequence>))
(for-each-with-index (^(i x)
(when (> i 0)
(display " "))
(display x))
seq)
(newline))
(define (prn . args)
(prn* args))
(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 ord char->integer)
(define chr integer->char)
(define int string->number)
(define str x->string)
(define string* (apply$ string))
(define-method min ((xs <sequence>))
(fold min (~ xs 0) xs))
(define-method max ((xs <sequence>))
(fold max (~ xs 0) xs))
(define (minmax . xs)
(values->list (apply min&max xs)))
(define-method minmax ((xs <sequence>))
(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 sq square)
(define isqrt exact-integer-sqrt)
(define pos? positive?)
(define neg? negative?)
(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 <sequence>))
(let1 ht (make-hash-table equal-comparator)
(for-each (^x (hash-table-update! ht x 1+ 0)) xs)
(new 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-method len ((coll <collection>))
(size-of coll))
(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 (accumulate xs)
(list->vector (scan-left + 0 xs)))
(define (accum xs)
(let ((vs (accumulate xs)))
(case-lambda
((l r) ; [l, r]
(- (~ vs r) (if (> l 0) (~ vs (1- l)) 0)))
((r) ; [0, r]
(~ vs r)))))
(define (string->set s)
(list->set eqv-comparator (string->list s)))
(define (difference xs ys)
(let ((excludes (list->set ys)))
(filter (^x (not (set-contains? excludes x))) xs)))
;; <hash-table>
(define (make-dict)
(new (make-hash-table equal-comparator)))
(define-method new ((self <hash-table>))
(match-lambda*
(('get key)
(hash-table-get self key 0))
(('get key default)
(hash-table-get self key default))
(('put! key value)
(hash-table-put! self key value))
(('update! key proc)
(hash-table-update! self key proc 0))
(('update! key proc default)
(hash-table-update! self key proc default))
(('keys)
(hash-table-keys self))
(('values)
(hash-table-values self))
(('contains? key)
(hash-table-contains? self key))
(('items)
(hash-table->alist self))
(('map fn)
(hash-table-map self fn))))
(define (alist->dict alist)
(new (alist->hash-table alist equal-comparator)))
;; '(1 2 3) -> op(op(op(init, a), b), c)
(define (scan-left op init xs)
(define (f a b)
(let1 t (op b a)
(values t t)))
(map-accum f init xs))
;;'(1 2 3) -> op(a, op(b, op(c, init)))
(define (scan-right op init xs)
(define (f a b)
(let1 t (op a b)
(values t t)))
(reverse (map-accum f init (reverse xs))))
(define (nth xs index)
(let loop ((p index)
(xs xs))
(if (zero? p)
(car xs)
(loop (1- p)
(cdr xs)))))
(input! ((N K))
(let* ((acc (scan-right (^(a b) (* (1+ a) b))
1
(digits N))))
(->
(let loop ((acc (map cons (digits N) acc))
(k (1- K))
(res '()))
(if (null? acc)
res
(mlet1 (d . a) (car acc)
(let* ((r (div a (1+ d))) ; 下位の桁のパターン数
(x (div k r)) ; 現在の桁を何番目の要素にするか
)
(loop (cdr acc)
(- k (* x r))
(cons x res))))))
reverse
digits->int
prn)))