結果

問題 No.1361 [Zelkova 4th Tune *] QUADRUPLE-SEQUENCEの詩
ユーザー sansaquasansaqua
提出日時 2021-01-22 23:45:07
言語 Common Lisp
(sbcl 2.3.8)
結果
AC  
実行時間 584 ms / 2,000 ms
コード長 14,736 bytes
コンパイル時間 919 ms
コンパイル使用メモリ 64,424 KB
実行使用メモリ 81,740 KB
最終ジャッジ日時 2023-08-27 23:28:21
合計ジャッジ時間 21,191 ms
ジャッジサーバーID
(参考情報)
judge11 / judge12
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
27,836 KB
testcase_01 AC 10 ms
23,948 KB
testcase_02 AC 10 ms
23,880 KB
testcase_03 AC 9 ms
25,760 KB
testcase_04 AC 9 ms
23,928 KB
testcase_05 AC 10 ms
27,884 KB
testcase_06 AC 10 ms
27,832 KB
testcase_07 AC 9 ms
23,792 KB
testcase_08 AC 10 ms
24,156 KB
testcase_09 AC 9 ms
23,844 KB
testcase_10 AC 10 ms
24,140 KB
testcase_11 AC 11 ms
27,468 KB
testcase_12 AC 10 ms
24,104 KB
testcase_13 AC 10 ms
25,944 KB
testcase_14 AC 9 ms
23,824 KB
testcase_15 AC 11 ms
24,128 KB
testcase_16 AC 11 ms
24,196 KB
testcase_17 AC 10 ms
27,468 KB
testcase_18 AC 33 ms
30,508 KB
testcase_19 AC 19 ms
26,504 KB
testcase_20 AC 32 ms
26,388 KB
testcase_21 AC 21 ms
30,436 KB
testcase_22 AC 24 ms
26,416 KB
testcase_23 AC 31 ms
30,460 KB
testcase_24 AC 32 ms
28,440 KB
testcase_25 AC 31 ms
30,512 KB
testcase_26 AC 33 ms
28,440 KB
testcase_27 AC 23 ms
26,544 KB
testcase_28 AC 222 ms
51,232 KB
testcase_29 AC 183 ms
47,120 KB
testcase_30 AC 152 ms
38,840 KB
testcase_31 AC 199 ms
45,112 KB
testcase_32 AC 276 ms
61,564 KB
testcase_33 AC 149 ms
42,440 KB
testcase_34 AC 265 ms
63,404 KB
testcase_35 AC 262 ms
47,036 KB
testcase_36 AC 424 ms
67,588 KB
testcase_37 AC 118 ms
42,912 KB
testcase_38 AC 198 ms
40,868 KB
testcase_39 AC 123 ms
38,932 KB
testcase_40 AC 168 ms
44,984 KB
testcase_41 AC 245 ms
51,116 KB
testcase_42 AC 362 ms
57,436 KB
testcase_43 AC 229 ms
47,104 KB
testcase_44 AC 271 ms
50,920 KB
testcase_45 AC 221 ms
40,896 KB
testcase_46 AC 257 ms
63,588 KB
testcase_47 AC 149 ms
42,944 KB
testcase_48 AC 499 ms
77,784 KB
testcase_49 AC 441 ms
77,876 KB
testcase_50 AC 508 ms
77,968 KB
testcase_51 AC 479 ms
77,848 KB
testcase_52 AC 448 ms
79,936 KB
testcase_53 AC 569 ms
79,912 KB
testcase_54 AC 555 ms
79,940 KB
testcase_55 AC 584 ms
81,740 KB
testcase_56 AC 559 ms
81,700 KB
testcase_57 AC 424 ms
79,976 KB
testcase_58 AC 104 ms
38,880 KB
testcase_59 AC 105 ms
38,876 KB
testcase_60 AC 103 ms
42,480 KB
testcase_61 AC 102 ms
40,908 KB
testcase_62 AC 102 ms
39,000 KB
testcase_63 AC 88 ms
42,596 KB
testcase_64 AC 88 ms
38,940 KB
testcase_65 AC 88 ms
38,856 KB
testcase_66 AC 88 ms
41,040 KB
testcase_67 AC 88 ms
38,884 KB
testcase_68 AC 9 ms
23,768 KB
testcase_69 AC 10 ms
25,852 KB
testcase_70 AC 10 ms
23,796 KB
testcase_71 AC 10 ms
25,836 KB
testcase_72 AC 9 ms
23,888 KB
testcase_73 AC 9 ms
23,904 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 27 AUG 2023 11:28:00 PM):
; processing (IN-PACKAGE :CL-USER)
; processing (DEFPARAMETER *OPT* ...)
; processing (SET-DISPATCH-MACRO-CHARACTER #\# ...)
; processing (SETQ *RANDOM-STATE* ...)
; processing (DEFINE-INT-TYPES 2 ...)
; processing (DEFCONSTANT +MOD+ ...)
; processing (DEFMACRO DBG ...)
; processing (DECLAIM (INLINE PRINTLN))
; processing (DEFUN PRINTLN ...)
; processing (DEFPACKAGE :CP/READ-FIXNUM ...)
; processing (IN-PACKAGE :CP/READ-FIXNUM)
; processing (DECLAIM (FTYPE # ...))
; processing (DEFUN READ-FIXNUM ...)
; processing (DEFPACKAGE :CP/BISECT ...)
; processing (IN-PACKAGE :CP/BISECT)
; processing (DECLAIM (INLINE BISECT-LEFT))
; processing (DEFUN BISECT-LEFT ...)
; processing (DECLAIM (INLINE BISECT-RIGHT))
; processing (DEFUN BISECT-RIGHT ...)
; processing (USE-PACKAGE :CP/BISECT ...)
; processing (USE-PACKAGE :CP/READ-FIXNUM ...)
; processing (IN-PACKAGE :CL-USER)
; processing (DEFUN SORT< ...)
; processing (DEFUN SORT> ...)
; processing (DEFUN SOLVE ...)
; file: /home/judge/data/code/Main.lisp
; in: DEFUN SOLVE
;     (>= (CALC-COUNT MID) S)
; 
; note: unable to open code because: The operands might not be the same type.
; 
; note: forced to do GENERIC-> (cost 10)
;       unable to do inline fixnum comparison (cost 4) because:
;       The second argument is a REAL, not a FIXNUM.

; processing (DEFUN MAIN ...)
; file: /home/judge/data/code/Main.lisp
; in: DEFUN MAIN
;     (MULTIPLE-VALUE-BIND (Y REM)
;         (FLOOR RES X)
;       (WHEN (AND (ZEROP REM) (GETHASH Y TABLE2))
;         (LOOP NAMED OUTER
;               FOR A ACROSS AS
;               DO (LOOP FOR B ACROSS BS
;                        WHEN (= # X)
;                        DO (FORMAT T "~D ~D " A B) (RETURN-FROM OUTER)))
;         (LOOP FOR C ACROSS CS
;               DO (LOOP FOR D ACROSS DS
;                        WHEN (= # Y)
;                        DO (FORMAT T "~D ~D~%" C D) (RETURN-FROM MAIN)))))
; ==>
;   (MULTIPLE-V

ソースコード

diff #

(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 (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/bisect
  (:use :cl)
  (:export #:bisect-left #:bisect-right))
(in-package :cp/bisect)

(declaim (inline bisect-left))
(defun bisect-left (target value &key (start 0) end (order #'<) (key #'identity))
  "TARGET := vector | function (taking an integer argument)
ORDER := strict order

Analogue of lower_bound() of C++ or bisect_left() of Python: Returns the
smallest index (or input) i that fulfills TARGET[i] >= VALUE, where '>=' is the
complement of ORDER. In other words, this function returns the leftmost index at
which VALUE can be inserted with keeping the order. Therefore, TARGET must be
monotonically non-decreasing with respect to ORDER.

- This function returns END if VALUE exceeds TARGET[END-1]. 
- The range [START, END) is half-open.
- END must be explicitly specified if TARGET is function.
- KEY is applied to each element of TARGET before comparison."
  (declare (integer start)
           ((or null integer) end))
  (macrolet
      ((frob (accessor &optional declaration)
         `(labels
              ((%bisect-left (ng ok)
                 ;; TARGET[OK] >= VALUE always holds (assuming
                 ;; TARGET[END] = +infinity)
                 ,@(when declaration (list declaration))
                 (if (<= (- ok ng) 1)
                     ok
                     (let ((mid (ash (+ ng ok) -1)))
                       (if (funcall order (funcall key (,accessor target mid)) value)
                           (%bisect-left mid ok)
                           (%bisect-left ng mid))))))
            (assert (<= start end))
            (%bisect-left (- start 1) end))))
    (etypecase target
      (vector
       (let ((end (or end (length target))))
         (frob aref (declare ((integer -1 (#.most-positive-fixnum)) ng ok)))))
      (function
       (assert end () "Requires END argument if TARGET is a function.")
       (frob funcall)))))

(declaim (inline bisect-right))
(defun bisect-right (target value &key (start 0) end (order #'<) (key #'identity))
  "TARGET := vector | function (taking an integer argument)
ORDER := strict order

Analogue of upper_bound() of C++ or bisect_right() of Python: Returns the
smallest index (or input) i that fulfills TARGET[i] > VALUE. In other words,
this function returns the rightmost index at which VALUE can be inserted with
keeping the order. Therefore, TARGET must be monotonically non-decreasing with
respect to ORDER.

- This function returns END if VALUE >= TARGET[END-1].
- The range [START, END) is half-open.
- END must be explicitly specified if TARGET is function.
- KEY is applied to each element of TARGET before comparison."
  (declare (integer start)
           ((or null integer) end))
  (macrolet
      ((frob (accessor &optional declaration)
         `(labels
              ((%bisect-right (ng ok)
                 ;; TARGET[OK] > VALUE always holds (assuming
                 ;; TARGET[END] = +infinity)
                 ,@(when declaration (list declaration))
                 (if (<= (- ok ng) 1)
                     ok
                     (let ((mid (ash (+ ng ok) -1)))
                       (if (funcall order value (funcall key (,accessor target mid)))
                           (%bisect-right ng mid)
                           (%bisect-right mid ok))))))
            (assert (<= start end))
            (%bisect-right (- start 1) end))))
    (etypecase target
      (vector
       (let ((end (or end (length target))))
         (frob aref (declare ((integer -1 (#.array-total-size-limit)) ng ok)))))
      (function
       (assert end () "Requires END argument if TARGET is a function.")
       (frob funcall)))))

;; BEGIN_USE_PACKAGE
(eval-when (:compile-toplevel :load-toplevel :execute)
  (use-package :cp/bisect :cl-user))
(eval-when (:compile-toplevel :load-toplevel :execute)
  (use-package :cp/read-fixnum :cl-user))
(in-package :cl-user)

;;;
;;; Body
;;;


(defun sort< (as)
  (declare #.*opt*
           (inline sort)
           ((simple-array int32 (*)) as))
  (sort as #'<))

(defun sort> (as)
  (declare #.*opt*
           (inline sort)
           ((simple-array int32 (*)) as))
  (sort as #'>))

(defun solve (as bs s)
  (declare #.*opt*
           ((simple-array int32 (*)) as bs))
  (let* ((n (length as))
         (m (length bs))
         (as+ (make-array n :fill-pointer 0 :element-type 'int32))
         (as- (make-array n :fill-pointer 0 :element-type 'int32))
         (bs+ (make-array m :fill-pointer 0 :element-type 'int32))
         (bs- (make-array m :fill-pointer 0 :element-type 'int32)))
    (declare (uint31 n m))
    (dotimes (i n)
      (let ((a (aref as i)))
        (assert (vector-push a (if (>= a 0) as+ as-)))))
    (dotimes (i m)
      (let ((b (aref bs i)))
        (assert (vector-push b (if (>= b 0) bs+ bs-)))))
    (let* ((as+< (sort< (coerce as+ '(simple-array int32 (*)))))
           (as+> (sort> (coerce as+ '(simple-array int32 (*)))))
           (as-< (sort< (coerce as- '(simple-array int32 (*)))))
           (as-> (sort> (coerce as- '(simple-array int32 (*)))))
           (bs+< (sort< (coerce bs+ '(simple-array int32 (*)))))
           (bs+> (sort> (coerce bs+ '(simple-array int32 (*)))))
           (bs-< (sort< (coerce bs- '(simple-array int32 (*)))))
           (bs-> (sort> (coerce bs- '(simple-array int32 (*))))))
      (declare ((simple-array int32 (*))
                as+< as+> as-< as->
                bs+< bs+> bs-< bs->))
      (labels ((calc-count (threshold)
                 (declare (fixnum threshold))
                 (let ((res 0))
                   (declare (uint62 res))
                   (when (>= threshold 0)
                     (let ((end 0))
                       (sb-int:dovector (a as+>)
                         (loop until (or (= end (length bs+<))
                                         (< threshold (* a (aref bs+< end))))
                               do (incf end)
                               finally (incf res end)))))
                   (if (>= threshold 0)
                       (incf res (the uint62 (* (length as+<) (length bs-<))))
                       (let ((end 0))
                         (sb-int:dovector (a as+<)
                           (loop until (or (= end (length bs-<))
                                           (< threshold (* a (aref bs-< end))))
                                 do (incf end)
                                 finally (incf res end)))))
                   (if (>= threshold 0)
                       (incf res (the uint62 (* (length as-<) (length bs+<))))
                       (let ((end 0))
                         (sb-int:dovector (a as->)
                           (loop until (or (= end (length bs+>))
                                           (< threshold (* a (aref bs+> end))))
                                 do (incf end)
                                 finally (incf res end)))))
                   (when (>= threshold 0)
                     (let ((end 0))
                       (sb-int:dovector (a as-<)
                         (loop until (or (= end (length bs->))
                                         (< threshold (* a (aref bs-> end))))
                               do (incf end)
                               finally (incf res end)))))
                   ;; (loop for a across as
                   ;;       for b across bs
                   ;;       when (<= (* a b)))
                   ;; (sb-int:dovector (a as)
                   ;;   (when (<= (* a a) threshold)
                   ;;     (decf res)))
                   ;; (setq res (ash res -1))
                   res)))
        (sb-int:named-let bisect ((ng #.(- (expt 10 18))) (ok #.(expt 10 18)))
          (declare (fixnum ng ok))
          (if (<= (- ok ng) 1)
              ok
              (let ((mid (floor (+ ng ok) 2)))
                (if (>= (calc-count mid) s)
                    (bisect ng mid)
                    (bisect mid ok)))))))))

(defun main ()
  (declare #.*opt*)
  (let* ((k (read))
         (l (read))
         (m (read))
         (n (read))
         (s (read))
         (as (make-array k :element-type 'int32 :initial-element 0))
         (bs (make-array l :element-type 'int32 :initial-element 0))
         (cs (make-array m :element-type 'int32 :initial-element 0))
         (ds (make-array n :element-type 'int32 :initial-element 0))
         (vec1 (make-array (* k l) :element-type 'int32 :initial-element 0))
         (vec2 (make-array (* m n) :element-type 'int32 :initial-element 0))
         (table2 (make-hash-table :test #'eq)))
    (declare (uint31 k l m n)
             (uint62 s))
    (dotimes (i k)
      (setf (aref as i) (read-fixnum)))
    (dotimes (i l)
      (setf (aref bs i) (read-fixnum)))
    (dotimes (i m)
      (setf (aref cs i) (read-fixnum)))
    (dotimes (i n)
      (setf (aref ds i) (read-fixnum)))
    (dotimes (i k)
      (dotimes (j l)
        (setf (aref vec1 (+ (* i l) j))
              (* (aref as i) (aref bs j)))))
    (dotimes (i m)
      (dotimes (j n)
        (setf (aref vec2 (+ (* i n) j))
              (* (aref cs i) (aref ds j)))))
    (let* ((res (solve vec1 vec2 s)))
      (declare (fixnum res))
      (sb-int:dovector (x vec2)
        (setf (gethash x table2) t))
      (println res)
      (if (zerop res)
          (multiple-value-bind (x1 x2 x3 x4) 
              (cond ((find 0 as) (values 0 (aref bs 0) (aref cs 0) (aref ds 0)))
                    ((find 0 bs) (values (aref as 0) 0 (aref cs 0) (aref ds 0)))
                    ((find 0 cs) (values (aref as 0) (aref bs 0) 0 (aref ds 0)))
                    ((find 0 ds) (values (aref as 0) (aref bs 0) (aref cs 0) 0))
                    (t (error "Huh?")))
            (format t "~D ~D ~D ~D~%" x1 x2 x3 x4))
          (loop for x across vec1
                unless (zerop x)
                do (multiple-value-bind (y rem) (floor res x)
                     (when (and (zerop rem) (gethash y table2))
                       (loop named outer
                             for a across as
                             do (loop for b across bs
                                      when (= (* a b) x)
                                      do (format t "~D ~D " a b)
                                         (return-from outer)))
                       (loop for c across cs
                             do (loop for d across ds
                                      when (= (* c d) y)
                                      do (format t "~D ~D~%" c d)
                                         (return-from main))))))))))

#-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/1361"))

#+swank
(defun gen-dat ()
  (uiop:with-output-file (out *dat-pathname* :if-exists :supersede)
    (format out "600 599 598 597 ~D~%" 1)
    (dotimes (_ 4)
      (dolist (len '(600 599 598 597))
        (dotimes (_ len)
          (println (- (random 60000) 30000) 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 "36
-1 2 -3 6
"
          (run "1 1 2 3 4
-1
2
-3 4
-5 6 -7
" nil)))
  (5am:is
   (equal "1
1 1 1 1
"
          (run "1 2 3 4 12
1
1 1
1 1 1
1 1 1 1
" nil)))
  (5am:is
   (equal "288
2 3 6 8
"
          (run "8 8 8 8 2000
3 1 4 1 5 9 2 6
5 3 5 8 9 7 9 3
2 3 8 4 6 2 6 4
3 3 8 3 2 7 9 5
" nil))))
0