結果

問題 No.1361 [Zelkova 4th Tune *] QUADRUPLE-SEQUENCEの詩
ユーザー sansaquasansaqua
提出日時 2021-01-22 23:43:11
言語 Common Lisp
(sbcl 2.3.8)
結果
AC  
実行時間 559 ms / 2,000 ms
コード長 14,737 bytes
コンパイル時間 1,683 ms
コンパイル使用メモリ 87,680 KB
実行使用メモリ 73,856 KB
最終ジャッジ日時 2024-06-08 18:58:35
合計ジャッジ時間 17,951 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
23,296 KB
testcase_01 AC 9 ms
23,296 KB
testcase_02 AC 9 ms
23,296 KB
testcase_03 AC 9 ms
23,296 KB
testcase_04 AC 9 ms
23,424 KB
testcase_05 AC 9 ms
23,296 KB
testcase_06 AC 9 ms
23,296 KB
testcase_07 AC 10 ms
23,296 KB
testcase_08 AC 10 ms
23,808 KB
testcase_09 AC 10 ms
23,552 KB
testcase_10 AC 10 ms
23,808 KB
testcase_11 AC 9 ms
23,680 KB
testcase_12 AC 10 ms
23,680 KB
testcase_13 AC 10 ms
23,552 KB
testcase_14 AC 10 ms
23,168 KB
testcase_15 AC 11 ms
23,680 KB
testcase_16 AC 10 ms
23,808 KB
testcase_17 AC 10 ms
23,552 KB
testcase_18 AC 28 ms
25,856 KB
testcase_19 AC 17 ms
24,448 KB
testcase_20 AC 30 ms
25,472 KB
testcase_21 AC 19 ms
25,600 KB
testcase_22 AC 22 ms
25,088 KB
testcase_23 AC 31 ms
25,984 KB
testcase_24 AC 30 ms
26,624 KB
testcase_25 AC 29 ms
25,984 KB
testcase_26 AC 31 ms
25,984 KB
testcase_27 AC 22 ms
24,576 KB
testcase_28 AC 217 ms
47,488 KB
testcase_29 AC 175 ms
40,832 KB
testcase_30 AC 150 ms
36,352 KB
testcase_31 AC 187 ms
42,240 KB
testcase_32 AC 254 ms
57,856 KB
testcase_33 AC 141 ms
36,224 KB
testcase_34 AC 255 ms
57,216 KB
testcase_35 AC 250 ms
42,752 KB
testcase_36 AC 408 ms
60,032 KB
testcase_37 AC 110 ms
39,936 KB
testcase_38 AC 182 ms
37,248 KB
testcase_39 AC 115 ms
33,152 KB
testcase_40 AC 158 ms
41,088 KB
testcase_41 AC 236 ms
48,000 KB
testcase_42 AC 339 ms
50,304 KB
testcase_43 AC 220 ms
42,112 KB
testcase_44 AC 250 ms
43,264 KB
testcase_45 AC 212 ms
38,016 KB
testcase_46 AC 246 ms
57,984 KB
testcase_47 AC 140 ms
40,192 KB
testcase_48 AC 445 ms
73,600 KB
testcase_49 AC 381 ms
73,728 KB
testcase_50 AC 451 ms
73,728 KB
testcase_51 AC 417 ms
73,728 KB
testcase_52 AC 407 ms
73,728 KB
testcase_53 AC 528 ms
73,728 KB
testcase_54 AC 507 ms
73,728 KB
testcase_55 AC 559 ms
73,728 KB
testcase_56 AC 508 ms
73,600 KB
testcase_57 AC 397 ms
73,856 KB
testcase_58 AC 102 ms
34,816 KB
testcase_59 AC 101 ms
34,560 KB
testcase_60 AC 101 ms
34,816 KB
testcase_61 AC 102 ms
34,688 KB
testcase_62 AC 98 ms
34,816 KB
testcase_63 AC 85 ms
34,688 KB
testcase_64 AC 84 ms
34,816 KB
testcase_65 AC 82 ms
34,816 KB
testcase_66 AC 86 ms
34,560 KB
testcase_67 AC 86 ms
34,816 KB
testcase_68 AC 9 ms
23,296 KB
testcase_69 AC 10 ms
23,296 KB
testcase_70 AC 9 ms
23,296 KB
testcase_71 AC 9 ms
23,424 KB
testcase_72 AC 9 ms
23,168 KB
testcase_73 AC 9 ms
23,296 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 08 JUN 2024 06:58:17 PM):

; wrote /home/judge/data/code/Main.fasl
; compilation finished in 0:00:00.250

ソースコード

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 ()
  (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)))
    (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)))
      (setq as (sort< as)
            bs (sort< bs)
            cs (sort< cs)
            ds (sort< ds))
      (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