結果
問題 | No.1443 Andd |
ユーザー | sansaqua |
提出日時 | 2021-03-26 22:47:35 |
言語 | Common Lisp (sbcl 2.3.8) |
結果 |
TLE
|
実行時間 | - |
コード長 | 15,927 bytes |
コンパイル時間 | 1,399 ms |
コンパイル使用メモリ | 62,724 KB |
実行使用メモリ | 66,544 KB |
最終ジャッジ日時 | 2024-11-29 00:44:28 |
合計ジャッジ時間 | 21,294 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 10 ms
36,236 KB |
testcase_01 | AC | 11 ms
66,044 KB |
testcase_02 | AC | 10 ms
36,232 KB |
testcase_03 | AC | 22 ms
66,544 KB |
testcase_04 | AC | 38 ms
36,620 KB |
testcase_05 | AC | 26 ms
62,120 KB |
testcase_06 | AC | 21 ms
38,532 KB |
testcase_07 | AC | 37 ms
64,284 KB |
testcase_08 | AC | 32 ms
36,232 KB |
testcase_09 | AC | 11 ms
29,540 KB |
testcase_10 | AC | 54 ms
31,692 KB |
testcase_11 | AC | 43 ms
31,584 KB |
testcase_12 | AC | 40 ms
31,580 KB |
testcase_13 | AC | 947 ms
32,064 KB |
testcase_14 | AC | 961 ms
38,140 KB |
testcase_15 | AC | 953 ms
32,312 KB |
testcase_16 | AC | 955 ms
34,096 KB |
testcase_17 | AC | 949 ms
32,196 KB |
testcase_18 | TLE | - |
testcase_19 | TLE | - |
testcase_20 | TLE | - |
testcase_21 | TLE | - |
testcase_22 | TLE | - |
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 29 NOV 2024 12:44:04 AM): ; wrote /home/judge/data/code/Main.fasl ; compilation finished in 0:00:00.147
ソースコード
(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 (dolist (f '(:popcnt :sse4)) (pushnew f sb-c:*backend-subfeatures*)) #+sbcl (setq *random-state* (seed-random-state (nth-value 1 (get-time-of-day))))) #-swank (eval-when (:compile-toplevel) (setq *break-on-signals* '(and warning (not style-warning)))) #+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+ 1000000007) (defmacro dbg (&rest forms) (declare (ignorable forms)) #+swank (if (= (length forms) 1) `(format *error-output* "~A => ~A~%" ',(car forms) ,(car forms)) `(format *error-output* "~A => ~A~%" ',forms `(,,@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 (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)))))))) (defpackage :cp/bit-basher (:use :cl) (:export #:bit-not! #:bit-fill! #:bit-count #:bit-lshift #:bit-rshift #:bit-shift) (:documentation "Provides several operations on bit vector that are not included in the standard.")) (in-package :cp/bit-basher) (eval-when (:compile-toplevel :load-toplevel :execute) (assert (= sb-vm:n-word-bits 64))) ;; KLUDGE: a variant of DPB that handles a 64-bit word efficiently (defmacro u64-dpb (new spec int) (destructuring-bind (byte s p) spec (assert (eql 'byte byte)) (let ((size (gensym)) (posn (gensym)) (mask (gensym))) `(let* ((,size ,s) (,posn ,p) (,mask (ldb (byte ,size 0) -1))) (logior (the (unsigned-byte 64) (ash (logand ,new ,mask) ,posn)) (the (unsigned-byte 64) (logand ,int (lognot (ash ,mask ,posn))))))))) (defconstant +most-positive-word+ (ldb (byte 64 0) -1)) (defun bit-not! (sb-vector &optional (start 0) end) "Destructively flips the bits in the range [START, END)." (declare (optimize (speed 3)) (simple-bit-vector sb-vector) ((mod #.array-total-size-limit) start) ((or null (mod #.array-total-size-limit)) end)) (setq end (or end (length sb-vector))) (assert (<= start end (length sb-vector))) (multiple-value-bind (start/64 start%64) (floor start 64) (multiple-value-bind (end/64 end%64) (floor end 64) (declare (optimize (safety 0))) (if (= start/64 end/64) (setf (sb-kernel:%vector-raw-bits sb-vector start/64) (u64-dpb (ldb (byte (- end%64 start%64) start%64) (logxor +most-positive-word+ (sb-kernel:%vector-raw-bits sb-vector start/64))) (byte (- end%64 start%64) start%64) (sb-kernel:%vector-raw-bits sb-vector start/64))) (progn (setf (sb-kernel:%vector-raw-bits sb-vector start/64) (dpb (sb-kernel:%vector-raw-bits sb-vector start/64) (byte start%64 0) (logxor +most-positive-word+ (sb-kernel:%vector-raw-bits sb-vector start/64)))) (loop for i from (+ 1 start/64) below end/64 do (setf (sb-kernel:%vector-raw-bits sb-vector i) (logxor +most-positive-word+ (sb-kernel:%vector-raw-bits sb-vector i)))) (unless (zerop end%64) (setf (sb-kernel:%vector-raw-bits sb-vector end/64) (dpb (logxor +most-positive-word+ (sb-kernel:%vector-raw-bits sb-vector end/64)) (byte end%64 0) (sb-kernel:%vector-raw-bits sb-vector end/64)))))))) sb-vector) (declaim (ftype (function * (values simple-bit-vector &optional)) bit-fill!)) (defun bit-fill! (sb-vector bit &optional (start 0) end) "Destructively sets the bits in the range [START, END) to BIT." (declare (optimize (speed 3)) (simple-bit-vector sb-vector) (bit bit) ((mod #.array-total-size-limit) start) ((or null (mod #.array-total-size-limit)) end)) (setq end (or end (length sb-vector))) (assert (<= start end (length sb-vector))) (let ((mask (if (zerop bit) 0 +most-positive-word+))) (multiple-value-bind (start/64 start%64) (floor start 64) (multiple-value-bind (end/64 end%64) (floor end 64) (if (= start/64 end/64) (setf (sb-kernel:%vector-raw-bits sb-vector start/64) (u64-dpb (ldb (byte (- end%64 start%64) 0) mask) (byte (- end%64 start%64) start%64) (sb-kernel:%vector-raw-bits sb-vector start/64))) (progn (setf (sb-kernel:%vector-raw-bits sb-vector start/64) (u64-dpb (sb-kernel:%vector-raw-bits sb-vector start/64) (byte start%64 0) mask)) (loop for i from (+ 1 start/64) below end/64 do (setf (sb-kernel:%vector-raw-bits sb-vector i) mask)) (unless (zerop end%64) (setf (sb-kernel:%vector-raw-bits sb-vector end/64) (dpb mask (byte end%64 0) (sb-kernel:%vector-raw-bits sb-vector end/64))))))))) sb-vector) ;; (count 1 simple-bit-vector) is sufficiently fast on SBCL when handling whole ;; vector. If START or END are specified, however, it is slow as the ;; deftransform for COUNT doesn't work. See ;; https://github.com/sbcl/sbcl/blob/cd7af0d5b15e98e21ace8ef164e0f39019e5ed4b/src/compiler/generic/vm-tran.lisp#L484-L527 (defun bit-count (sb-vector &optional (start 0) end) "Counts 1's in the range [START, END)." (declare (optimize (speed 3)) (simple-bit-vector sb-vector) ((mod #.array-total-size-limit) start) ((or null (mod #.array-total-size-limit)) end)) (setq end (or end (length sb-vector))) (assert (<= start end (length sb-vector))) (multiple-value-bind (start/64 start%64) (floor start 64) (multiple-value-bind (end/64 end%64) (floor end 64) (declare (optimize (safety 0))) (if (= start/64 end/64) (logcount (ldb (byte (- end%64 start%64) start%64) (sb-kernel:%vector-raw-bits sb-vector start/64))) (let ((result 0)) (declare ((mod #.array-total-size-limit) result)) (incf result (logcount (ldb (byte (- 64 start%64) start%64) (sb-kernel:%vector-raw-bits sb-vector start/64)))) (loop for i from (+ 1 start/64) below end/64 do (incf result (logcount (sb-kernel:%vector-raw-bits sb-vector i)))) (unless (zerop end%64) (incf result (logcount (ldb (byte end%64 0) (sb-kernel:%vector-raw-bits sb-vector end/64))))) result))))) (declaim (ftype (function * (values simple-bit-vector &optional)) bit-lshift)) (defun bit-lshift (bit-vector delta &optional result-vector end) "Left-shifts BIT-VECTOR by DELTA bits and fills the new bits with zero. The behaviour is the same as the bit-wise operations in ANSI CL: The result is copied to RESULT-VECTOR; if it is T, BIT-VECTOR is destructively modified; if it is NIL, a new bit-vector of the same length is created. If END is specified, this function shifts only the range [0, END) of BIT-VECTOR and copies it to the range [0, END+DELTA) of RESULT-VECTOR. Note that here `left' means the direction from a smaller index to a larger one and is contrary to the `visual' direction: i.e. (bit-lshift #*1011000 2) |-> #*0010110." (declare (optimize (speed 3)) (simple-bit-vector bit-vector) ((or null (eql t) simple-bit-vector) result-vector) ((mod #.array-total-size-limit) delta) ((or null (mod #.array-total-size-limit)) end)) (setq result-vector (etypecase result-vector (null (make-array (length bit-vector) :element-type 'bit :initial-element 0)) ((eql t) bit-vector) (simple-bit-vector result-vector))) (setq end (or end (length bit-vector))) (assert (<= end (length bit-vector))) (replace result-vector bit-vector :start1 (min (length result-vector) delta) :start2 0 :end2 end) (bit-fill! result-vector 0 0 (min delta (length result-vector)))) (declaim (ftype (function * (values simple-bit-vector &optional)) bit-rshift)) (defun bit-rshift (bit-vector delta &optional result-vector) "Right-shifts BIT-VECTOR by DELTA bits and fills the new bits with zero. The behaviour is the same as the bit-wise operations in ANSI CL: The result is copied to RESULT-VECTOR; if it is T, BIT-VECTOR is destructively modified; if it is NIL, a new bit-vector of the same length is created. Note that here `right' means the direction from a larger index to a smaller one and is contrary to the `visual' direction: i.e. (bit-rshift #*1011000 2) |-> #*1100000." (declare (optimize (speed 3)) (simple-bit-vector bit-vector) ((or null (eql t) simple-bit-vector) result-vector) ((mod #.array-total-size-limit) delta)) (setq result-vector (etypecase result-vector (null (make-array (length bit-vector) :element-type 'bit :initial-element 0)) ((eql t) bit-vector) (simple-bit-vector result-vector))) (replace result-vector bit-vector :start2 (min delta (length bit-vector))) (bit-fill! result-vector 0 (min (max 0 (- (length bit-vector) delta)) (length result-vector)))) ;; not tested (declaim (ftype (function * (values simple-bit-vector &optional)) bit-shift)) (defun bit-shift (bit-vector delta &optional result-vector) (declare (optimize (speed 3)) (simple-bit-vector bit-vector) ((or null (eql t) simple-bit-vector) result-vector) ((integer #.(- array-total-size-limit) #.array-total-size-limit) delta)) (if (>= delta 0) (bit-lshift bit-vector delta result-vector) (bit-rshift bit-vector (- delta) result-vector))) ;; (defun bit-rotate (bit-vector delta &optional result-vector) ;; (declare (optimize (speed 3)) ;; ((mod #.array-total-size-limit) delta) ;; (simple-bit-vector bit-vector) ;; ((or null simple-bit-vector) result-vector)) ;; (assert (not (eql bit-vector result-vector))) ;; (let* ((end (length bit-vector)) ;; (result-vector (or result-vector (make-array end :element-type 'bit))) ;; (delta (mod delta end))) ;; :unfinished)) (defpackage :cp/tzcount (:use :cl) (:export #:tzcount)) (in-package :cp/tzcount) (declaim (inline tzcount)) (defun tzcount (x) "Returns the number of trailing zero bits. Note that (TZCOUNT 0) = -1." (- (integer-length (logand x (- x))) 1)) (defpackage :cp/modify-macro (:use :cl) (:export #:minf #:maxf #:mulf #:divf #:iorf #:xorf #:andf)) (in-package :cp/modify-macro) (macrolet ((def (name fname) `(define-modify-macro ,name (new-value) ,fname))) (def minf min) (def maxf max) (def mulf *) (def divf /) (def iorf logior) (def xorf logxor) (def andf logand)) ;; BEGIN_USE_PACKAGE (eval-when (:compile-toplevel :load-toplevel :execute) (use-package :cp/modify-macro :cl-user)) (eval-when (:compile-toplevel :load-toplevel :execute) (use-package :cp/tzcount :cl-user)) (eval-when (:compile-toplevel :load-toplevel :execute) (use-package :cp/bit-basher :cl-user)) (eval-when (:compile-toplevel :load-toplevel :execute) (use-package :cp/read-fixnum :cl-user)) (in-package :cl-user) ;;; ;;; Body ;;; (defun main () (declare #.*opt*) (let* ((n (read)) (table (make-array #.(* 20000 1024) :element-type 'bit :initial-element 0)) (tmp (make-array #.(* 20000 1024) :element-type 'bit :initial-element 0)) (mask (make-array 1024 :element-type 'bit :initial-element 0))) (declare (uint31 n) (simple-bit-vector table tmp)) (setf (aref table 0) 1) (write-string (with-output-to-string (*standard-output* nil :element-type 'base-char) (dotimes (i n) (fill mask 0) (let* ((a (read-fixnum))) (declare (uint31 a)) (bit-lshift table a tmp (min (length tmp) (* (+ i 1) 1024))) (dotimes (j (+ i 1)) (let ((base (* j 16))) (loop for k below 16 do (iorf (sb-kernel:%vector-raw-bits mask k) (sb-kernel:%vector-raw-bits table (+ base k)))))) (dotimes (x (length mask)) (when (= 1 (aref mask x)) (iorf (aref tmp (logand x a)) 1))) (println (bit-count tmp 0 (min (length tmp) (* 1024 (+ i 2))))) (bit-fill! table 0 0 (* (+ i 1) 1024)) (rotatef table tmp))))))) #-swank (main) ;;; ;;; Test ;;; #+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/1443")) #+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 sb-c::*undefined-warnings* (error "undefined warnings: ~{~A~^ ~}" sb-c::*undefined-warnings*))) ;; To run: (5am:run! :sample) #+swank (5am:test :sample (5am:is (equal "2 3 " (run "2 1 4 " nil))) (5am:is (equal "2 4 5 9 " (run "4 30 29 16 26 " nil))) (5am:is (equal "2 4 6 6 9 17 22 34 60 63 " (run "10 26 7 10 0 10 30 21 30 31 20 " nil))))