#1111602 (Scheme) No.2092 Conjugation

• 問題: No.2092 Conjugation
• ユーザー: norioc
• 提出日時: 2025-08-10 18:15:11
• 言語: Scheme (Gauche-0.9.15)
• 結果: AC
• 実行時間: 544 ms / 2,000 ms
• コード長: 7,644 bytes
• コンパイル時間: 286 ms
• コンパイル使用メモリ: 7,968 KB
• 実行使用メモリ: 48,020 KB
• 最終ジャッジ日時: 2025-08-10 18:15:21
• 合計ジャッジ時間: 8,625 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

ファイルパターン	結果
sample	AC * 2
other	AC * 18

ソースコード

(use srfi.13)  ; string
(use srfi.42)  ; list-ec
(use srfi.197) ; chain
(use scheme.list)
(use scheme.vector)
(use data.queue)
(use gauche.dictionary)
(use gauche.generator)
(use gauche.sequence)
(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 <collection>))
  (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-syntax let/g
  (syntax-rules ()
    ((_ var body ...)
     (generate (lambda (var) body ...)))))

(define-method reverse ((s <string>))
  (string-reverse s))

(define-method reverse ((vs <vector>))
  (rlet1 res (vector-copy vs)
    (vector-reverse! res)))

(define ord char->integer)
(define chr integer->char)

(define int string->number)
(define str x->string)
(define string* (apply$ string))

(define-method min ((xs <collection>))
  (fold min (~ xs 0) xs))
(define-method max ((xs <collection>))
  (fold max (~ xs 0) xs))

(define (minmax . xs)
  (values->list (apply min&max xs)))
(define-method minmax ((xs <collection>))
  (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 == equal?)

(define (cmp< a b) (< (compare a b) 0))
(define (cmp> a b) (> (compare a b) 0))

(define ++ string-append)

(define all every)
(define any-ec any?-ec)
(define all-ec every?-ec)
(define concat concatenate)
(define foldl fold-left)

(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 <collection>))
  (let1 ht (make-hash-table equal-comparator)
    (for-each (^x (hash-table-update! ht x 1+ 0)) xs)
    (new ht)))

;; '(a b c) -> op(op(op(init, a), b), c)
(define (scanl op init xs)
  (define (f a b)
    (let1 t (op b a)
      (values t t)))

  (map-accum f init xs))

;; '(a b c) ->  op(a, op(b, op(c, init)))
(define (scanr op init xs)
  (define (f a b)
    (let1 t (op a b)
      (values t t)))

  (reverse (map-accum f init (reverse xs))))

(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 (readln bindings . body)
  (let* ((vs (gensym))
         (binds (map-with-index
                 (^(i bind)
                   (let ((var (car bind))
                         (type (cadr bind)))
                     (ecase type
                       ((int) `(,var (int (list-ref ,vs ,i))))
                       ((str) `(,var (list-ref ,vs ,i))))))
                 bindings)))
    `(let* ((,vs (string-split (input) " "))
            ,@binds)
       ,@body)))

(define-macro (! self quoted-name . args)
  (let ((name (cadr quoted-name)))
    `(,name ,self ,@args)))

(define-macro (d/ . args)
  (let ((ss (map (^x
                   `(list (quote ,x) ,x))
                 args)))
    `(pprint (list ,@ss))))

(define mlet match-let)
(define mlet* match-let*)
(define mlet1 match-let1)

(define-macro (mfn pat . body)
  (let ((args (gensym)))
    `(lambda ,args
       (mlet1 ,pat ,args
         ,@body))))

(define-macro (mfn1 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 equal-comparator ys)))
    (filter (^x (not (set-contains? excludes x))) xs)))

(define (uniq xs)
  (let1 ht (make-hash-table 'equal?)
    (for-each (^x (hash-table-put! ht x #t))
              xs)
    (hash-table-keys ht)))

(define (replace-index-accessor var tree)
  (let loop ((tree tree)
             (res '()))
    (if (null? tree)
      (if (= 1 (length res))
        (car res)
        (reverse res))
      (let ((hd (car tree)))
        (cond
         ((eq? hd var)
          (loop (cddr tree)
                (cons `(~ ,var ,(replace-index-accessor var (cadr tree))) res)))
         ((pair? hd)
          (loop (cdr tree)
                (cons (replace-index-accessor var hd) res)))
         (else
          (loop (cdr tree)
                (cons hd res))))))))

(define-macro (expand-rhs var . exprs)
  (assume (symbol? var))
  (replace-index-accessor var exprs))

(define-syntax ?
  (syntax-rules (= += <?= >?=)
;;    ((_ var (index) = expr ...)
;;     (let1 rhs (expand-rhs var expr ...)
;;       (set! (~ var index) rhs)))

    ((_ var (l-expr ...) = expr ...)
     (let1 rhs (expand-rhs var expr ...)
       (set! (~ var (l-expr ...)) rhs)))

    ((_ var (index) += expr ...)
     (let1 rhs (expand-rhs var expr ...)
       (inc! (~ var index) rhs)))

    ((_ var (l-expr ...) += expr ...)
     (let1 rhs (expand-rhs var expr ...)
       (inc! (~ var (l-expr ...)) rhs)))

    ((_ var (index) <?= expr ...)
     (let1 rhs (expand-rhs var expr ...)
       (set! (~ var index) (min (~ var index) rhs))
       ))

    ((_ var (index) >?= expr ...)
     (let1 rhs (expand-rhs var expr ...)
       (set! (~ var index) (max (~ var index) rhs))
       ))
    ))


(input! (N (A))
  (let1 vs (make-vector (car A) 0)
    (dolist (x (map 1- A))
      (? vs[x] += 1))

    (->
     (scanr + 0 vs)
     prn*)))