結果

問題 No.196 典型DP (1)
ユーザー sansaquasansaqua
提出日時 2021-12-01 20:08:49
言語 Common Lisp
(sbcl 2.3.8)
結果
AC  
実行時間 46 ms / 2,000 ms
コード長 7,035 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 49,344 KB
実行使用メモリ 46,564 KB
最終ジャッジ日時 2023-09-18 04:39:13
合計ジャッジ時間 2,697 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 10 ms
23,500 KB
testcase_01 AC 10 ms
23,576 KB
testcase_02 AC 11 ms
27,604 KB
testcase_03 AC 10 ms
23,568 KB
testcase_04 AC 11 ms
27,636 KB
testcase_05 AC 10 ms
23,564 KB
testcase_06 AC 10 ms
23,512 KB
testcase_07 AC 10 ms
23,488 KB
testcase_08 AC 9 ms
23,568 KB
testcase_09 AC 11 ms
27,560 KB
testcase_10 AC 10 ms
27,540 KB
testcase_11 AC 10 ms
23,564 KB
testcase_12 AC 10 ms
23,532 KB
testcase_13 AC 10 ms
23,608 KB
testcase_14 AC 10 ms
23,592 KB
testcase_15 AC 11 ms
23,732 KB
testcase_16 AC 13 ms
25,840 KB
testcase_17 AC 15 ms
23,960 KB
testcase_18 AC 17 ms
24,188 KB
testcase_19 AC 19 ms
24,160 KB
testcase_20 AC 24 ms
26,284 KB
testcase_21 AC 23 ms
26,384 KB
testcase_22 AC 23 ms
24,352 KB
testcase_23 AC 31 ms
30,548 KB
testcase_24 AC 30 ms
34,244 KB
testcase_25 AC 28 ms
32,080 KB
testcase_26 AC 46 ms
46,564 KB
testcase_27 AC 44 ms
42,924 KB
testcase_28 AC 44 ms
42,904 KB
testcase_29 AC 45 ms
42,948 KB
testcase_30 AC 44 ms
44,780 KB
testcase_31 AC 34 ms
32,476 KB
testcase_32 AC 35 ms
32,484 KB
testcase_33 AC 34 ms
32,508 KB
testcase_34 AC 35 ms
32,492 KB
testcase_35 AC 34 ms
32,468 KB
testcase_36 AC 33 ms
32,524 KB
testcase_37 AC 24 ms
28,360 KB
testcase_38 AC 34 ms
32,564 KB
testcase_39 AC 32 ms
32,464 KB
testcase_40 AC 36 ms
36,340 KB
testcase_41 AC 10 ms
25,604 KB
testcase_42 AC 10 ms
23,552 KB
testcase_43 AC 10 ms
23,564 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 18 SEP 2023 04:39:08 AM):
; processing (IN-PACKAGE :CL-USER)
; processing (DEFPARAMETER *OPT* ...)
; processing (SET-DISPATCH-MACRO-CHARACTER #\# ...)
; processing (DOLIST (F #) ...)
; 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/MOD-OPERATIONS ...)
; processing (IN-PACKAGE :CP/MOD-OPERATIONS)
; processing (DEFVAR *MODULUS* ...)
; processing (DECLAIM (# *MODULUS*) ...)
; processing (DEFMACRO DEFINE-MOD-OPERATIONS ...)
; processing (DEFINE-MOD-OPERATIONS COMMON-LISP-USER::+MOD+ ...)
; processing (USE-PACKAGE :CP/MOD-OPERATIONS ...)
; processing (IN-PACKAGE :CL-USER)
; processing (DEFUN MAIN ...)
; file: /home/judge/data/code/Main.lisp
; in: DEFUN MAIN
;     (AREF DP K)
; 
; note: unable to
;   optimize
; because:
;   Upgraded element type of array is not known at compile time.

;     (INCF SIZE (DFS CHILD V))
; --> SETQ THE 
; ==>
;   (+ (DFS CHILD V) SIZE)
; 
; note: unable to
;   open-code float conversion in mixed numeric operation
; due to type uncertainty:
;   The second argument is a NUMBER, not a FLOAT.
; 
; note: unable to
;   open-code float conversion in mixed numeric operation
; due to type uncertainty:
;   The second argument is a NUMBER, not a (OR (COMPLEX SINGLE-FLOAT)
;                                              (COMPLEX DOUBLE-FLOAT)).
; 
; 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 &REST T).
;       unable to do inline (signed-byte 64) arithmetic (cost 4) because:
;       The second argument is a NUMBER, not a (SIGNED-BYTE 64).
;       The result is a (VALUES NUMBER &OPTIONAL), not a (VALUES (SIGNED-BYTE 64)

ソースコード

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 (dolist (f '(:popcnt :sse4)) (pushnew f sb-c:*backend-subfeatures*))
  (setq *random-state* (make-random-state t)))
#-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/mod-operations
  (:use :cl)
  (:export #:define-mod-operations #:*modulus*)
  (:documentation "Provides modular arithmetic."))
(in-package :cp/mod-operations)

(eval-when (:compile-toplevel :load-toplevel :execute)
  (defvar *modulus* 0))
(declaim ((unsigned-byte 31) *modulus*)
         #+sbcl (sb-ext:always-bound *modulus*))

(defmacro define-mod-operations
    (divisor &optional (package #+sbcl (sb-int:sane-package) #-sbcl *package*))
  (let ((mod* (intern "MOD*" package))
        (mod+ (intern "MOD+" package))
        (mod- (intern "MOD-" package))
        (incfmod (intern "INCFMOD" package))
        (decfmod (intern "DECFMOD" package))
        (mulfmod (intern "MULFMOD" package)))
    `(progn
       (defun ,mod* (&rest args)
         (cond ((cdr args) (reduce (lambda (x y) (mod (* x y) ,divisor)) args))
               (args (mod (car args) ,divisor))
               (t (mod 1 ,divisor))))
       (defun ,mod+ (&rest args)
         (cond ((cdr args) (reduce (lambda (x y) (mod (+ x y) ,divisor)) args))
               (args (mod (car args) ,divisor))
               (t 0)))
       (defun ,mod- (&rest args)
         (if (cdr args)
             (reduce (lambda (x y) (mod (- x y) ,divisor)) args)
             (mod (- (car args)) ,divisor)))

       #+sbcl
       (eval-when (:compile-toplevel :load-toplevel :execute)
         (locally (declare (sb-ext:muffle-conditions warning))
           (sb-c:define-source-transform ,mod* (&rest args)
             (case (length args)
               (0 `(mod 1 ,',divisor))
               (1 `(mod ,(car args) ,',divisor))
               (otherwise (reduce (lambda (x y) `(mod (* ,x ,y) ,',divisor)) args))))
           (sb-c:define-source-transform ,mod+ (&rest args)
             (case (length args)
               (0 0)
               (1 `(mod ,(car args) ,',divisor))
               (otherwise (reduce (lambda (x y) `(mod (+ ,x ,y) ,',divisor)) args))))
           (sb-c:define-source-transform ,mod- (&rest args)
             (case (length args)
               (0 (values nil t))
               (1 `(mod (- ,(car args)) ,',divisor))
               (otherwise (reduce (lambda (x y) `(mod (- ,x ,y) ,',divisor)) args))))))

       (define-modify-macro ,incfmod (delta)
         (lambda (x y) (mod (+ x y) ,divisor)))
       (define-modify-macro ,decfmod (delta)
         (lambda (x y) (mod (- x y) ,divisor)))
       (define-modify-macro ,mulfmod (multiplier)
         (lambda (x y) (mod (* x y) ,divisor))))))

(define-mod-operations cl-user::+mod+ :cl-user)

;; BEGIN_USE_PACKAGE
(eval-when (:compile-toplevel :load-toplevel :execute)
  (use-package :cp/mod-operations :cl-user))
(in-package :cl-user)

;;;
;;; Body
;;;

(defun main ()
  (declare (optimize (speed 3)))
  (let* ((n (read))
         (k (read))
         (graph (make-array n :element-type 'list :initial-element nil))
         (sizes (make-array n :element-type 'uint31 :initial-element 0)))
    (dotimes (_ (- n 1))
      (let ((u (read))
            (v (read)))
        (push u (aref graph v))
        (push v (aref graph u))))
    (sb-int:named-let dfs ((v 0) (parent -1))
      (let ((size 1))
        (dolist (child (aref graph v))
          (unless (eql child parent)
            (incf size (dfs child v))))
        (setf (aref sizes v) size)))
    (let ((dp
            (sb-int:named-let dfs ((v 0) (parent -1))
              (let ((dp (make-array 1 :element-type 'uint31 :initial-element 1)))
                (declare ((simple-array uint31 (*)) dp))
                (dolist (child (aref graph v))
                  (unless (eql child parent)
                    (let* ((cdp (dfs child v))
                           (new-dp (make-array (+ (length cdp) (length dp) -1)
                                               :element-type 'uint31
                                               :initial-element 0)))
                      (declare ((simple-array uint31 (*)) cdp))
                      (dotimes (i (length dp))
                        (dotimes (j (length cdp))
                          (incfmod (aref new-dp (+ i j))
                                   (mod* (aref dp i) (aref cdp j)))))
                      (setq dp new-dp))))
                (adjust-array dp (+ 1 (aref sizes v)) :initial-element 1)))))
      (println (aref dp k)))))

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

#+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 "3
"
          (run "6 1
0 1
0 2
1 3
2 4
2 5
" nil)))
  (5am:is
   (equal "4
"
          (run "6 2
0 1
0 2
1 3
2 4
2 5
" nil)))
  (5am:is
   (equal "4
"
          (run "6 3
0 1
0 2
1 3
2 4
2 5
" nil))))
0