結果

問題 No.917 Make One With GCD
ユーザー sansaqua
提出日時 2020-11-15 09:06:22
言語 Common Lisp
(sbcl 2.5.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 3,471 bytes
コンパイル時間 172 ms
コンパイル使用メモリ 46,072 KB
実行使用メモリ 23,552 KB
最終ジャッジ日時 2024-07-23 01:03:37
合計ジャッジ時間 1,444 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 23 JUL 2024 01:03:35 AM):

; file: /home/judge/data/code/Main.lisp
; in: DEFUN MAIN
;     (INCF (GETHASH A (AREF DP X) 0))
; ==>
;   (+ 1 (GETHASH #:KEY #:HASHTABLE 0))
; 
; note: unable to
;   associate +/(+ -) of constants
; due to type uncertainty:
;   The first argument is a NUMBER, not a RATIONAL.

;     (GCD DIV (AREF AS DEST))
; 
; note: unable to
;   optimize
; due to type uncertainty:
;   The first argument is a (UNSIGNED-BYTE 62), not a (OR
;                                                      (INTEGER
;                                                       -4611686018427387904 -1)
;                                                      (INTEGER 1
;                                                       4611686018427387903)).
;   The second argument is a (UNSIGNED-BYTE 31), not a (OR
;                                                       (INTEGER
;                                                        -4611686018427387904 -1)
;                                                       (INTEGER 1
;                                                        4611686018427387903)).

;     (INCF (GETHASH A (AREF DP X) 0))
; ==>
;   (+ 1 (GETHASH #:KEY #:HASHTABLE 0))
; 
; note: forced to do GENERIC-+ (cost 10)
;       unable to do inline fixnum arithmetic (cost 1) because:
;       The first argument is a NUMBER, not a FIXNUM.
;       The result is a (VALUES NUMBER &OPTIONAL), not a (VALUES FIXNUM &OPTIONAL).
;       unable to do inline fixnum arithmetic (cost 2) because:
;       The first argument is a NUMBER, not a FIXNUM.
;       The result is a (VALUES NUMBER &OPTIONAL), not a (VALUES FIXNUM &OPTIONAL).
;       etc.

;     (INCF (GETHASH GCD (AREF DP DEST) 0) NUM)
; 
; note: forced to do GENERIC-+ (cost 10)
;       unable to do inline fixnum arithmetic (cost 2) because:
;       The second argument is a NUMBER, not a FIXNUM.
;       The result is a (VALUES NUMBER &OPTIONAL), not a (VALUES FIXNUM &OPTIONAL).

ソースコード

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)))))
#+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)
  #+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* 'double-float))
    (prog1 (princ obj stream) (terpri stream))))

;; BEGIN_INSERTED_CONTENTS
;; BEGIN_USE_PACKAGE
(in-package :cl-user)

;;;
;;; Body
;;;

(defun main ()
  (declare #.*opt*)
  (let* ((n (read))
         (as (make-array n :element-type 'uint31 :initial-element 0))
         (dp (make-array n :element-type 'hash-table)))
    (dotimes (i n)
      (setf (aref as i) (read)
            (aref dp i) (make-hash-table :size 1000 :test #'eq)))
    (dotimes (x n)
      (let ((a (aref as x)))
        (incf (gethash a (aref dp x) 0))
        (loop for dest from (+ x 1) below n
              do (maphash (lambda (div num)
                            (declare (uint62 div num))
                            (let ((gcd (gcd div (aref as dest))))
                              (incf (gethash gcd (aref dp dest) 0) num)))
                          (aref dp x)))))
    (let ((res 0))
      (dotimes (x n)
        (incf res (gethash 1 (aref dp x) 0)))
      (println res))))

#-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*))
  (defparameter *dat-pathname* (uiop:merge-pathnames* "test.dat" *lisp-file-pathname*))
  (defparameter *problem-url* "https://yukicoder.me/problems/no/917"))

#+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)))

#-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 "5
"
          (run "3
1 2 3
" nil)))
  (5am:is
   (equal "15
"
          (run "4
1 1 1 1
" nil)))
  (5am:is
   (equal "0
"
          (run "4
2 4 8 16
" nil)))
  (5am:is
   (equal "763
"
          (run "10
801754 703742 332182 68016 914814 8470 937255 293192 313080 501971
" nil))))
0