Submission #409108

---

Source Code Expand

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

(in-package :king-collection)

(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)))

(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