結果

問題 No.1972 Modulo Set
コンテスト
ユーザー n_get
提出日時 2022-07-16 16:16:19
言語 Scheme
(Gauche-0.9.15)
結果
AC  
実行時間 352 ms / 2,000 ms
コード長 2,899 bytes
コンパイル時間 273 ms
コンパイル使用メモリ 5,760 KB
実行使用メモリ 29,952 KB
最終ジャッジ日時 2024-06-28 15:48:20
合計ジャッジ時間 7,742 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

(use util.queue)(use util.match)(use gauche.collection)(use gauche.sequence)
(define-syntax def-syntax (syntax-rules()((_ v e ...)(define-syntax v (syntax-rules e ...)))))
(define-syntax def-const define-constant)(define-syntax def define)(define-syntax def-inline define-inline)
(def-syntax def-names ()((_[x y]...)(begin(define-constant x y)...)))
(def-names [// quotient][% modulo][v-make make-vector][v-ref vector-ref][v-set! vector-set!][ht-make make-hash-table][ht-ref hash-table-get][ht-set! hash-table-put!][ht-keys hash-table-keys][ht-values hash-table-values][q-make make-queue][make-q make-queue][enq! enqueue!][deq! dequeue!][q-empty? queue-empty?])

(def-syntax input (: list vec cons str)
  ((_ : cons e ...)(cons(input : e ...)(input : e ...)))((_ : list str)(input : string->list str))
  ((_ : vec str)(input : string->vector str))((_ : vec n e ...)(vector-tabulate n(^_(input : e ...))))
  ((_ : list n e ...)(let l([k n])(if(zero? k)'()(cons(input : e ...)(l(- k 1))))))((_ :)(read))((_ : : e)e)
  ((_ : str)(let1 s(read-line)(if(string=? s "")(read-line)s)))((_[[v]: e ...])(def v(input : e ...)))
  ((_[[v w ...]: e ...])(begin(def v(input : e ...))(input[[w ...]: e ...])))
  ((_[v : e ...])(def v(input : e ...)))((_ : op e ...)(op(input : e ...)))
  ((_(e ...))(e ...))((_ v)(def v(read)))((_ e ...)(begin(input e)...)))

(def-syntax for [in to with]
  ((_[i N]e ...)(dotimes [i N] e ...))
  ((_[i A to B]e ...)
   (let[[a A][b B]](if (< A B)(let loop[[i a]] e ... (when (< i b) (loop (+ i 1))))
                       (let loop[[i a]] e ... (when (> i b) (loop (- i 1)))))))
  ((_[[x ...]in[A ...]with i]e ...)(for-each-with-index(^(i x ...)e ...)A ...))
  ((_[[x ...]in[A ...]]e ...)(for-each(^(x ...)e ...)A ...))
  ((_[x in A with i]e ...)(for-each-with-index(^(i x)e ...)A))
  ((_[x in A]e ...)(for-each(^(x) e ...)A))
  ((_[k v in H]e ...)(hash-table-for-each H(^(k v)e ...))))


(define-constant *MOD* 1000000007)
;(define-constant *MOD* 998244353)
(def-syntax add-mod! ()((_ v x)(let1 y(+ v x)(set! v(if(< y *MOD*)y(- y *MOD*))))))
(def-inline(v-add-mod! v i x)(let1 y(+(v-ref v i)x)(v-set! v i(if(< y *MOD*)y(- y *MOD*)))))
(def-syntax max! ()((_ v x)(set! v (max x v)))) (def-syntax min! ()((_ v x)(set! v (min x v))))
(def-inline(v-add! v i x)(v-set! v i(+ x(v-ref v i))))
(def-inline(ht-add! ht i x)(ht-set! ht i(+ x(ht-ref ht i 0))))
(def-inline(ht-cons! ht i x)(ht-set! ht i(cons x(ht-ref ht i '()))))
(def-syntax return () ((_ e ...) (begin (print e ...) (exit))))


;; 以下スニペットのコピペ


;; 以下が本体

(input N M [A : vec N])
(def ht (make-hash-table))
(for [a in A]
  (ht-add! ht (mod a M) 1)
  (ht-add! ht (- M (mod a M)) 0))
(def ans 0)
(for [a n in ht]
  (when (< 0 a (- M a))
    (inc! ans (max n (ht-ref ht (- M a))))))
(when (even? M)
  (inc! ans (min 1 (ht-ref ht (ash M -1) 0))))
(inc! ans (min 1 (ht-ref ht 0 0)))

(print ans)
0