快速幂解斐波那契数列

题目描述

写一个函数，输入 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。