結果
問題 | No.1299 Random Array Score |
ユーザー | sansaqua |
提出日時 | 2020-11-27 21:31:50 |
言語 | Common Lisp (sbcl 2.5.0) |
結果 |
AC
|
実行時間 | 72 ms / 2,000 ms |
コード長 | 7,533 bytes |
コンパイル時間 | 1,374 ms |
コンパイル使用メモリ | 52,184 KB |
実行使用メモリ | 42,112 KB |
最終ジャッジ日時 | 2024-07-23 22:14:04 |
合計ジャッジ時間 | 4,466 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 23 JUL 2024 10:13:59 PM): ; wrote /home/judge/data/code/Main.fasl ; compilation finished in 0:00:00.365
ソースコード
(in-package :cl-user)(eval-when (:compile-toplevel :load-toplevel :execute)(defparameter *opt*#+swank '(optimize (speed 3) (safety 2))#-swank '(optimize (speed 3) (safety 0) (debug 0)))#+swank (ql:quickload '(:cl-debug-print :fiveam :cp/util) :silent t)#+swank (use-package :cp/util :cl-user)#-swank (set-dispatch-macro-character#\# #\> (lambda (s c p) (declare (ignore c p)) `(values ,(read s nil nil t))))#+sbcl (setq *random-state* (seed-random-state (nth-value 1 (get-time-of-day)))))#+swank (set-dispatch-macro-character #\# #\> #'cl-debug-print:debug-print-reader)(macrolet ((def (b)`(progn (deftype ,(intern (format nil "UINT~A" b)) () '(unsigned-byte ,b))(deftype ,(intern (format nil "INT~A" b)) () '(signed-byte ,b))))(define-int-types (&rest bits) `(progn ,@(mapcar (lambda (b) `(def ,b)) bits))))(define-int-types 2 4 7 8 15 16 31 32 62 63 64))(defconstant +mod+ 998244353)(defmacro dbg (&rest forms)#+swank (if (= (length forms) 1)`(format *error-output* "~A => ~A~%" ',(car forms) ,(car forms))`(format *error-output* "~A => ~A~%" ',forms `(,,@forms)))#-swank (declare (ignore forms)))(declaim (inline println))(defun println (obj &optional (stream *standard-output*))(let ((*read-default-float-format*(if (typep obj 'double-float) 'double-float *read-default-float-format*)))(prog1 (princ obj stream) (terpri stream))));; BEGIN_INSERTED_CONTENTS;;;;;; Arithmetic operations with static modulus;;;(defpackage :cp/mod-operations(:use :cl)(:export #:define-mod-operations))(in-package :cp/mod-operations)(defmacro define-mod-operations(divisor &optional (package #+sbcl (sb-int:sane-package) #-sbcl *package*))(let ((mod* (intern "MOD*" package))(mod+ (intern "MOD+" package))(mod- (intern "MOD-" package))(incfmod (intern "INCFMOD" package))(decfmod (intern "DECFMOD" package))(mulfmod (intern "MULFMOD" package)))`(progn(defun ,mod* (&rest args)(cond ((cdr args) (reduce (lambda (x y) (mod (* x y) ,divisor)) args))(args (mod (car args) ,divisor))(t 1)))(defun ,mod+ (&rest args)(cond ((cdr args) (reduce (lambda (x y) (mod (+ x y) ,divisor)) args))(args (mod (car args) ,divisor))(t 0)))(defun ,mod- (&rest args)(if (cdr args)(reduce (lambda (x y) (mod (- x y) ,divisor)) args)(mod (- (car args)) ,divisor)))#+sbcl(eval-when (:compile-toplevel :load-toplevel :execute)(locally (declare (sb-ext:muffle-conditions warning))(sb-c:define-source-transform ,mod* (&rest args)(case (length args)(0 1)(1 `(mod ,(car args) ,',divisor))(otherwise (reduce (lambda (x y) `(mod (* ,x ,y) ,',divisor)) args))))(sb-c:define-source-transform ,mod+ (&rest args)(case (length args)(0 0)(1 `(mod ,(car args) ,',divisor))(otherwise (reduce (lambda (x y) `(mod (+ ,x ,y) ,',divisor)) args))))(sb-c:define-source-transform ,mod- (&rest args)(case (length args)(0 (values nil t))(1 `(mod (- ,(car args)) ,',divisor))(otherwise (reduce (lambda (x y) `(mod (- ,x ,y) ,',divisor)) args))))))(define-modify-macro ,incfmod (delta)(lambda (x y) (mod (+ x y) ,divisor)))(define-modify-macro ,decfmod (delta)(lambda (x y) (mod (- x y) ,divisor)))(define-modify-macro ,mulfmod (multiplier)(lambda (x y) (mod (* x y) ,divisor))))))(define-mod-operations cl-user::+mod+ :cl-user)(defpackage :cp/mod-power(:use :cl)(:export #:mod-power))(in-package :cp/mod-power)(declaim (inline mod-power))(defun mod-power (base power modulus)"Returns BASE^POWER mod MODULUS. Note: 0^0 = 1.BASE := integerPOWER, MODULUS := non-negative fixnum"(declare ((integer 0 #.most-positive-fixnum) modulus power)(integer base))(let ((base (mod base modulus))(res (mod 1 modulus)))(declare ((integer 0 #.most-positive-fixnum) base res))(loop while (> power 0)when (oddp power)do (setq res (mod (* res base) modulus))do (setq base (mod (* base base) modulus)power (ash power -1)))res))(defpackage :cp/read-fixnum(:use :cl)(:export #:read-fixnum))(in-package :cp/read-fixnum)(declaim (ftype (function * (values fixnum &optional)) read-fixnum))(defun read-fixnum (&optional (in *standard-input*))"NOTE: cannot read -2^62"(macrolet ((%read-byte ()`(the (unsigned-byte 8)#+swank (char-code (read-char in nil #\Nul))#-swank (sb-impl::ansi-stream-read-byte in nil #.(char-code #\Nul) nil))))(let* ((minus nil)(result (loop (let ((byte (%read-byte)))(cond ((<= 48 byte 57)(return (- byte 48)))((zerop byte) ; #\Nul(error "Read EOF or #\Nul."))((= byte #.(char-code #\-))(setq minus t)))))))(declare ((integer 0 #.most-positive-fixnum) result))(loop(let* ((byte (%read-byte)))(if (<= 48 byte 57)(setq result (+ (- byte 48)(* 10 (the (integer 0 #.(floor most-positive-fixnum 10))result))))(return (if minus (- result) result))))))));; BEGIN_USE_PACKAGE(eval-when (:compile-toplevel :load-toplevel :execute)(use-package :cp/read-fixnum :cl-user))(eval-when (:compile-toplevel :load-toplevel :execute)(use-package :cp/mod-power :cl-user))(eval-when (:compile-toplevel :load-toplevel :execute)(use-package :cp/mod-operations :cl-user))(in-package :cl-user);;;;;; Body;;;(defun main ()(let* ((n (read))(k (read)))(println (mod* (reduce #'mod+ (loop repeat n collect (read-fixnum)))(mod-power 2 k +mod+)))))#-swank (main);;;;;; Test and benchmark;;;#+swank(progn(defparameter *lisp-file-pathname* (uiop:current-lisp-file-pathname))(setq *default-pathname-defaults* (uiop:pathname-directory-pathname *lisp-file-pathname*))(uiop:chdir *default-pathname-defaults*)(defparameter *dat-pathname* (uiop:merge-pathnames* "test.dat" *lisp-file-pathname*))(defparameter *problem-url* "https://yukicoder.me/problems/no/1299"))#+swank(defun gen-dat ()(uiop:with-output-file (out *dat-pathname* :if-exists :supersede)(format out "")))#+swank(defun bench (&optional (out (make-broadcast-stream)))(time (run *dat-pathname* out)))#+(and sbcl (not swank))(eval-when (:compile-toplevel)(when (or (> sb-c::*compiler-warning-count* 0)sb-c::*undefined-warnings*)(error "count: ~D, undefined warnings: ~A"sb-c::*compiler-warning-count*sb-c::*undefined-warnings*)));; To run: (5am:run! :sample)#+swank(5am:test :sample(5am:is(equal "12"(run "3 11 2 3" nil)))(5am:is(equal "1"(run "1 01" nil)))(5am:is(equal "461591775"(run "10 10000000000000000003 1 4 1 5 9 2 6 5 3" nil))))