斐波那契(矩阵实现)

大家都知道斐波那契数列,现在要求输入一个正整数 n ,请你输出斐波那契数列的第 n 项。

斐波那契数列是一个满足 image-20211003113124484 的数列

数据范围:n<=39

这里只讲如何利用矩阵去计算
[ 1 0 0 1 1 1 1 0 1 2 0 1 2 3 1 1 1 0 3 5 1 2 0 1 5 8 2 3 1 1 1 0 8 13 3 5 1 2 0 1 ] \begin{bmatrix} 1 & 0 \\ 0 & 1 \\ 1 & 1 & 1 & 0 \\ 1 & 2 & 0 & 1 \\ 2 & 3 & 1 & 1 & 1 & 0 \\ 3 & 5 & 1 & 2 & 0 & 1 \\ 5 & 8 & 2 & 3 & 1 & 1 & 1 & 0 \\ 8 & 13 & 3 & 5 & 1 & 2 & 0 & 1 \\ \end{bmatrix} 10112358011235813101123011235101101121001
我们不难看出这是一个分块矩阵

且我们可以得到以下规律
A = [ 1 0 0 1 ] A =\begin{bmatrix} 1 & 0 \\ 0 & 1 \\ \end{bmatrix} A=[1001]

B = [ 1 1 1 2 ] B = \begin{bmatrix} 1 & 1\\ 1 & 2\\ \end{bmatrix} B=[1112]

F ( n ) = B n ∗ A F(n) = B^n*A F(n)=BnA

其中n为输入的数除2再向上取整并减1

这里矩阵存储的是fib(1)、fib(2)…的个数。

代码实现:

# -*- coding:utf-8 -*-
class Solution:
    def Fibonacci(self, n):
        if n == 0:
            return 0
        a = [[1, 0], [0, 1]]
        b = [[1, 1], [1, 2]]
        f = 1
        m = n // 2 - 1
        if n % 2 != 0:
            m += 1
            f = 0
        for i in range(m):
            t = [[0, 0], [0, 0]]
            t[0][0] = b[0][0] * a[0][0] + b[0][1] * a[1][0]
            t[0][1] = b[0][0] * a[0][1] + b[0][1] * a[1][1]
            t[1][0] = b[1][0] * a[0][0] + b[1][1] * a[1][0]
            t[1][1] = b[1][0] * a[0][1] + b[1][1] * a[1][1]
            a = t
        return a[f][0]*1+a[f][1]*1

时间复杂度:O(log(n))

空间复杂度:O(1)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值