結果

問題 No.2007 Arbitrary Mod (Easy)
ユーザー n_getn_get
提出日時 2022-07-15 22:07:44
言語 Scheme
(Gauche-0.9.15)
結果
AC  
実行時間 40 ms / 2,000 ms
コード長 2,865 bytes
コンパイル時間 441 ms
コンパイル使用メモリ 5,888 KB
実行使用メモリ 19,072 KB
最終ジャッジ日時 2024-06-27 18:10:42
合計ジャッジ時間 4,994 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

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)...)))

(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))))

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

(define-constant MOD 1000000007)
(define-inline (expt-mod a n)
  (let loop ([k 1] [ak a] [ls '()])
    (if (<= k n) (loop (* k 2) (remainder (* ak ak) MOD) (cons (cons k ak) ls))
        (let loop2 ([ls ls] [n n] [an 1])
          (cond [(zero? n) an]
                [(> (caar ls) n) (loop2 (cdr ls) n an)]
                [else (loop2 (cdr ls) (- n (caar ls)) (remainder (* an (cdar ls)) MOD))])))))
(define-inline (inverse-mod a)
  (expt-mod a (- MOD 2)))
(define-inline (fact-mod n)
  (let loop([n n] [s 1]) (if (zero? n) s (loop (- n 1) (remainder (* s n) MOD)))))
(define-inline (nPr-mod n r)
  (let loop([n n] [r r] [s 1]) (if (zero? r) s (loop (- n 1) (- r 1) (remainder (* s n) MOD)))))
(define-inline (nCr-mod n r)
  (let ([R (min r (- n r))])
    (remainder (* (nPr-mod n R) (inverse-mod (fact-mod R))) MOD)))

;; 以下が本体
(input a n)
(print MOD)
(print (expt-mod a n))
0