結果

問題 No.917 Make One With GCD
ユーザー sansaqua
提出日時 2020-11-15 11:03:32
言語 Common Lisp
(sbcl 2.5.0)
結果
AC  
実行時間 13 ms / 2,000 ms
コード長 3,540 bytes
コンパイル時間 422 ms
コンパイル使用メモリ 40,036 KB
実行使用メモリ 32,880 KB
最終ジャッジ日時 2024-07-23 01:08:40
合計ジャッジ時間 1,647 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 23 JUL 2024 01:08:38 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*
          (if (typep obj 'double-float) 'double-float *read-default-float-format*)))
    (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