快速幂解斐波那契数列

题目描述

写一个函数,输入 n ,求斐波那契(Fibonacci)数列的第 n 项(即 F(N))。斐波那契数列的定义如下:

F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。

答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。

示例 1:

输入:n = 2
输出:1
示例 2:

输入:n = 5
输出:5

提示:

0 <= n <= 100

解答思路

正常递归会超时,所以需要使用数学计算方法

快速幂
在这里插入图片描述
在这里插入图片描述

最后输出上图的左上角即可

提交结果

在这里插入图片描述

package main

import "fmt"

func main() {

    fmt.Println(fib(95))
}

func fib(n int) int {
    //异常一超出时间限制
    //异常二求和顺序和计算顺序错误,先将数组二的值算出来,就会先运算,计算错误,不能将值为一的时候的值计算出来,一直都是值为二的
    //异常三矩阵计算的时候power2[0][0]先算出来后,算power2[0][1]会计算出错   两次
    //异常四应该是每次赋值为平方,不是每次和一相乘
    //异常五结果之间应该是相乘,不是想加
    //异常六提交的时候注意取模
    //异常七整数溢出,需要提前多次取模
    power1 := [][]int{{1, 1}, {1, 0}}
    power2 := [][]int{{1, 0}, {0, 1}}
    if n == 0 {
        return 0
    }
    if n == 1 {
        return 1
    }
    m := n - 1
    var p00, p01, p10, p11 int
    for m/2 != 0 || m == 1 {
        if m%2 == 1 {
            p00 = power2[0][0]*power1[0][0] + power2[0][1]*power1[1][0]
            p01 = power2[0][0]*power1[0][1] + power2[0][1]*power1[1][1]
            p10 = power2[1][0]*power1[0][0] + power2[1][1]*power1[1][0]
            p11 = power2[1][0]*power1[0][1] + power2[1][1]*power1[1][1]
            power2[0][0] = p00 % 1000000007
            power2[0][1] = p01 % 1000000007
            power2[1][0] = p10 % 1000000007
            power2[1][1] = p11 % 1000000007
        }
        p00 = power1[0][0]*power1[0][0] + power1[0][1]*power1[1][0]
        p01 = power1[0][0]*power1[0][1] + power1[0][1]*power1[1][1]
        p10 = power1[1][0]*power1[0][0] + power1[1][1]*power1[1][0]
        p11 = power1[1][0]*power1[0][1] + power1[1][1]*power1[1][1]
        power1[0][0] = p00 % 1000000007
        power1[0][1] = p01 % 1000000007
        power1[1][0] = p10 % 1000000007
        power1[1][1] = p11 % 1000000007
        m /= 2
    }
    return power2[0][0] % 1000000007
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/fei-bo-na-qi-shu-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 1024 设计师:白松林 返回首页