斐波那契数快速算法

定义

$fib(n)=\{\begin{array}{cc}0& ifn=0\\ 1& ifn=1\\ fib(n-2)+fib(n-1)& otherwise\end{array}$

一般计算方法

直接用递归实现

def fib(n):
	if n == 0:
		return 0
	elif n == 1:
		return 1
	else:
		return fib(n-2) + fib(n-1)

时间复杂度为$O({\phi }^n)$，空间复杂度为$O(n)$，其中$\phi$为黄金分割率。

使用迭代实现
（1）

def fib(n):
	a, b = 0, 1
	for i in range(n):
		a, b = b, a + b
	return a

（2）

def fib(n):
	a, b = 0, 1
	while n:
		a, b = b, a + b
		n -= 1
	return a

时间复杂度为$O(n)$，空间复杂度为$O(n)$。

高效算法

考虑上面的迭代计算过程，迭代计算的核心思想是状态变换。
初始状态
$a=0$
$b=1$

状态更新
$a\leftarrow b$
$b\leftarrow a+b$

状态更新$n$次即可得到$fib(n)$和$fib(n+1)$。
这种状态更新形式事实上是线性变换，因此可用矩阵乘法表示。

$\left(\begin{array}{c}b\\ a\end{array}\right)\leftarrow \left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right]\left(\begin{array}{c}b\\ a\end{array}\right)$

其中
$T=\left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right]$
为线性变换矩阵。

令
$s_n=\left(\begin{array}{c}fib(n+1)\\ fib(n)\end{array}\right)$
则
$s_0=\left(\begin{array}{c}1\\ 0\end{array}\right)$

$s_n=T^n{s}_0$
可根据此事实设计高效算法

$s_n=T^n{s}_0=\{\begin{array}{cc}(T^2{)}^{\frac{n}{2}}{s}_0& ifniseven\\ T^{n-1}(T{s}_0)& ifnisodd\end{array}$

def fib(n)
	s0, s1 = 0, 1				#fib(0),fib(1)
	a, b, c, d = 1, 1, 1, 0		#变换矩阵
	while n:
		if n % 2 == 0:
			#n为偶数时，T<----T^2
			a, b, c, d = a**2 + b*c, b*(a + d), c*(a + d), b*c + d**2 
			n \= 2
		else:
			s0, s1 = c*s1 + d*s2, a*s1 + b*s2
			n -= 1
	return s0			

该算法的时间复杂度为$O(logn)$，空间复杂度为$O(1)$。

这里共使用了7个状态变量：n, s0, s1, a, b, c, d。考虑到变换矩阵$T$为实对称矩阵，有b=c，因此可以减少一个状态变量。

def fib(n)
	s0, s1 = 0, 1				#fib(0),fib(1)
	a, b, c = 1, 1, 0		#变换矩阵
	while n:
		if n % 2 == 0:
			#n为偶数时，T<----T^2
			a, b, c = a**2 + b**2, b*(a + c), b**2 + c**2 
			n \= 2
		else:
			s0, s1 = b*s1 + c*s2, a*s1 + b*s2
			n -= 1
	return s0			

再考虑到，$c=a-b$，变换矩阵只用两个变量a,b表示即可。

def fib(n)
	s0, s1 = 0, 1				#fib(0),fib(1)
	a, b = 1, 1		#变换矩阵
	while n:
		if n % 2 == 0:
			#n为偶数时，T<----T^2
			a, b= a**2 + b**2, b*(a + (a - b))
			n \= 2
		else:
			s0, s1 = b*s1 + (a-b)*s2, a*s1 + b*s2
			n -= 1
	return s0