Fast Fibonacci Number Matrix Calculation - psholtz/MIT-SICP GitHub Wiki
Exercise 1.19
There is a clever algorithm for computing the Fibonacci numbers in a logarithmic number of steps. Recall the transformation of the state variables a and b in the fib-iter process of Section 1.2.2: a <- a + b, and b <- a. Call this transformation T, and observe that applying T over and over again n times, starting with 1 and 0, produces the pair Fib(n+1) and Fib(n). In other words, the Fibonacci numbers are produced by applying T^n, the n-th power of the transformation T, starting with the pair (1,0). Now consider T to be the special case of p=0 and q=1 in a family of transformations T(pq), where T(pq) transforms the pair (a,b) according to a <- bq + aq + ap and b <- bp + aq. Show that if we apply such a transformation T(pq) twice, the effect is the same as using a single transformation T(p'q') of the same form, and compute p' and q' in terms of p and q. This gives us an explicit way to square these transformations, and thus we can compute T^n using successive squaring, as in the fast-expt procedure. Put this all together to complete the following procedure, which runs in a logarithmic number of steps:
(define (fib n)
(define (fib-iter a b p q count)
(cond ((= count 0) b)
((even? count)
(fib-iter a
b
<??> ;; compute p'
<??> ;; compute q'
(/ count 2)))
(else (fib-iter (+ (* b q) (* a q) (* a p))
(+ (* b p) (* a q))
p
q
(- count 1))))))
(fib-iter 1 0 0 1 n))
Solution
The linear operator for calculating Fibonacci numbers is given by:
Applying A n times will produce Fib(n+1) and Fib(n), to wit:
For instance:
and so on.
The question asks us to consider the more general linear transformation:
Clearly, the Fibonacci transformation is a special case of this more general transformation T, wherein p = 0 and q = 1.
We square T to obtain:
so that we have:
where
In other words, the transformation T^2 still preserves the same "form" as the original transformation T.
Knowing this, we can now rewrite our Fibonacci procedure in Scheme as follows:
(define (fib n)
(fib-iter 1 0 0 1 n))
(define (fib-iter a b p q count)
(cond ((= count 0) b)
((even? count)
(fib-iter a
b
(+ (square p) (square q))
(+ (* 2 p q) (square q))
(/ count 2)))
(else (fib-iter (+ (* b q) (* a q) (* a p))
(+ (* b p) (* a q))
p
q
(- count 1)))))
This procedure is easily able to calculate relatively large Fibonacci numbers.
For instance, evaluating (fib 100) results in 354224848179261915075, whereas this evaluation would hang an interpreter using a more "naive" implementation of (fib n).