racket-random/bst.rkt
joachimschmidt557 fade1fc218 WIP bst
2019-09-13 19:06:46 +02:00

60 lines
No EOL
1.9 KiB
Racket

#lang racket
(define-struct node (value left right))
(define (search node value)
(cond
[(= (node-value node) value) node]
[(empty? node) empty]
[(> (node-value node) value) (search (node-left node) value)]
[else (search (node-right node) value)]))
(define (insert node value)
(cond
[(empty? node) (make-node value empty empty)]
[(> (node-value node) value) (make-node (node-value node) (insert (node-left node) value) (node-right node))]
[else (make-node (node-value node) (node-left node) (insert (node-right node) value))]))
(define (bst-from-list lst)
(cond
[(empty? lst) empty]
[else (insert (bst-from-list (rest lst)) (first lst))]))
(define (rotate-right node)
(if (empty? node) empty
(let ([x node] [l (node-left node)] [r (node-right node)])
(let ([a (if (empty? l) empty (node-left l))]
[b (if (empty? l) empty (node-right l))]
[c (if (empty? r) empty (node-left r))])
(make-node (node-value l) a (make-node (node-value x) b r))))))
(define (height node)
(cond
[(empty? node) 0]
[else (+ 1 (max (height (node-left node)) (height (node-right node))))]))
(define (avlcheck node)
(cond
[(empty? node) 0]
[(= (avlcheck (node-left node)) -1) -1]
[(= (avlcheck (node-right node)) -1) -1]
[(< 1 (abs (- (avlcheck (node-left node)) (avlcheck (node-right node))))) -1]
[else (+ 1 (max (avlcheck (node-left node)) (avlcheck (node-right node))))]))
(define (min-subtree node)
(cond
[(empty? node) empty]
[(empty? (node-left node)) node]
[else (min-subtree (node-left node))]))
(define (max-subtree node)
(cond
[(empty? node) empty]
[(empty? (node-right node)) node]
[else (max-subtree (node-right node))]))
(define (delete node)
(cond
[(empty? node) empty]
[(empty? (node-left node)) (node-right node)]
[(empty? (node-right node)) (node-left node)]))