77 lines
3.1 KiB
Scheme
77 lines
3.1 KiB
Scheme
;;; The SRFI-?? sort package -- stable vector insertion sort -*- Scheme -*-
|
|
;;; Copyright (c) 1998 by Olin Shivers.
|
|
;;; This code is open-source; see the end of the file for porting and
|
|
;;; more copyright information.
|
|
;;; Olin Shivers 10/98.
|
|
|
|
;;; Exports:
|
|
;;; insert-sort < v [start end] -> vector
|
|
;;; insert-sort! < v [start end] -> unspecific
|
|
;;;
|
|
;;; %insert-sort! is also called from vqsort.scm's quick-sort function.
|
|
|
|
(define (insert-sort elt< v . maybe-start+end)
|
|
(call-with-values
|
|
(lambda () (vector-start+end v maybe-start+end))
|
|
(lambda (start end)
|
|
(let ((ans (vector-portion-copy v start end)))
|
|
(%insert-sort! elt< ans 0 (- end start))
|
|
ans))))
|
|
|
|
(define (insert-sort! < v . maybe-start+end)
|
|
(call-with-values
|
|
(lambda () (vector-start+end v maybe-start+end))
|
|
(lambda (start end)
|
|
(%insert-sort! < v start end))))
|
|
|
|
(define (%insert-sort! elt< v start end)
|
|
(do ((i (+ 1 start) (+ i 1))) ; Invariant: [start,i) is sorted.
|
|
((>= i end))
|
|
(let ((val (vector-ref v i)))
|
|
(vector-set! v (let lp ((j i)) ; J is the location of the
|
|
(if (<= j start)
|
|
start ; "hole" as it bubbles down.
|
|
(let* ((j-1 (- j 1))
|
|
(vj-1 (vector-ref v j-1)))
|
|
(cond ((elt< val vj-1)
|
|
(vector-set! v j vj-1)
|
|
(lp j-1))
|
|
(else j)))))
|
|
val))))
|
|
|
|
;;; Copyright
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
;;; This code is
|
|
;;; Copyright (c) 1998 by Olin Shivers.
|
|
;;; The terms are: You may do as you please with this code, as long as
|
|
;;; you do not delete this notice or hold me responsible for any outcome
|
|
;;; related to its use.
|
|
;;;
|
|
;;; Blah blah blah. Don't you think source files should contain more lines
|
|
;;; of code than copyright notice?
|
|
|
|
|
|
;;; Code tuning & porting
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
;;;
|
|
;;; This code is tightly bummed as far as I can go in portable Scheme.
|
|
;;;
|
|
;;; The code can be converted to use unsafe vector-indexing and
|
|
;;; fixnum-specific arithmetic ops -- the safety checks done on entry to
|
|
;;; INSERT-SORT and INSERT-SORT! are sufficient to guarantee nothing bad will
|
|
;;; happen. However, note that if you alter %INSERT-SORT! to use dangerous
|
|
;;; primitives, you must ensure it is only called from clients that guarantee
|
|
;;; to observe its preconditions. In the SRFI-?? reference implementation,
|
|
;;; %INSERT-SORT! is only called from INSERT-SORT! and the quick-sort code in
|
|
;;; vqsort.scm, and the preconditions are guaranteed for these two clients.
|
|
;;; This should provide *big* speedups. In fact, all the code bumming I've
|
|
;;; done pretty much disappears in the noise unless you have a good compiler
|
|
;;; and also can dump the vector-index checks and generic arithmetic -- so
|
|
;;; I've really just set things up for you to exploit.
|
|
;;;
|
|
;;; If your Scheme has a faster mechanism for handling optional arguments
|
|
;;; (e.g., Chez), you should definitely port over to it. Note that argument
|
|
;;; defaulting and error-checking are interleaved -- you don't have to
|
|
;;; error-check defaulted START/END args to see if they are fixnums that are
|
|
;;; legal vector indices for the corresponding vector, etc.
|