Submission #409106

---

Source Code Expand

(defpackage king-collection
  (:use COMMON-LISP)
  (:export king-collection4))

(in-package :king-collection)

(declaim (inline from-string-to-number))
(defun from-string-to-number ()
  (declare (optimize (speed 3) (safety 0) (debug 0)))
  (let ((ss (concatenate 'vector (read-line)))
	(oo '())
	(ee '()))
    (loop for i from 0 to (- (array-dimension ss 0) 1) collect
	 (if (not (eq #\  (aref ss i)))
	     (setf oo (cons (aref ss i) oo))
	     (progn
	       (setf ee (cons (concatenate 'string (reverse oo)) ee))
	       (setf oo nil))))
    (setf ee (cons (concatenate 'string (reverse oo)) ee))
    (setf oo nil)
    (setf ee (reverse ee))
    (mapcar (lambda (x) (parse-integer x :junk-allowed t)) ee)))

(declaim (inline king-collection4))
(defun king-collection4 ()
  (declare (optimize (speed 3) (debug 0) (safety 0)))
  (let* ((r-c-k (from-string-to-number))
	 (r (nth 0 r-c-k))
	 (c (nth 1 r-c-k))
	 (k (nth 2 r-c-k))
	 (n (read))
	 (r-r (make-array '(100001)))
	 (c-c (make-array '(100001)))
	 (r-r-r (make-array '(100001)))
	 (c-c-c (make-array '(100001)))
	 (result 0)
	 (vr (make-array 0
			 :fill-pointer 0
			 :adjustable t))
	 (vc (make-array 0
			 :fill-pointer 0
			 :adjustable t)))
    
    (loop for i from 0 to (- n 1) by 1 collect
	 (let ((a (read)) (b (read)))
	   (decf a)
	   (decf b)
	   (incf (svref r-r a))
	   (incf (svref c-c b))
	   (vector-push-extend a vr)
	   (vector-push-extend b vc)))
    
    (loop for i from 0 to (- c 1) by 1 collect
	 (incf (svref c-c-c (svref c-c i))))
    
    (loop for i from 0 to (- r 1 ) by 1 collect
	 (incf (svref r-r-r (svref r-r i))))
    
    (loop for i from 0 to k by 1 collect
	 (setf result (+ result (* (svref r-r-r i) (svref c-c-c (- k i))))))
    
    (loop for i from 0 to (- (length vr) 1) by 1 collect
	 (let* ((rr (svref r-r (aref vr i)))
		(cc (svref c-c (aref vc i)))
		(sum (+ rr cc)))
	   (if (= sum k)
	       (decf result)
	       (if (= sum (+ k 1))
		   (incf result)))))
    
    (format t "~a~%" result)))

(king-collection4)

Submission Info

Judge Result