1
1
mirror of https://github.com/kanaka/mal.git synced 2024-08-16 17:20:23 +03:00

examples: add memoize.mal as a usage example for atoms

This commit is contained in:
Dov Murik 2015-12-10 12:50:36 -05:00
parent a136ea497e
commit a624ba4b67

53
examples/memoize.mal Normal file
View File

@ -0,0 +1,53 @@
;;
;; memoize.mal
;;
;; Impelement `memoize` using an atom (`mem`) which holds the memoized results
;; (hash-map from the arguments to the result). When the function is called,
;; the hash-map is checked to see if the result for the given argument was already
;; calculated and stored. If this is the case, it is returned immediately;
;; otherwise, it is calculated and stored in `mem`.
;;
;; Adapated from http://clojure.org/atoms
;;
;; Memoize any function
(def! memoize
(fn* [f]
(let* [mem (atom {})]
(fn* [& args]
(let* [key (str args)]
(if (contains? @mem key)
(get @mem key)
(let* [ret (apply f args)]
(do
(swap! mem assoc key ret)
ret))))))))
;; Naive (non-memoized) Fibonacci function
(def! fib
(fn* [n]
(if (<= n 1)
n
(+ (fib (- n 1)) (fib (- n 2))))))
;; -----------------------------------------------
;; Benchmarks
(load-file "../perf.mal") ; for the 'time' macro
(def! N 32)
;; Benchmark naive 'fib'
(println "fib N=" N ": without memoization:")
(time (fib N))
;; "Elapsed time: 14402 msecs"
;; Benchmark memoized 'fib'
(def! fib (memoize fib))
(println "fib N=" N ": with memoization:")
(time (fib N))
;; "Elapsed time: 1 msecs"