斐波那契数列加强版 快速矩阵幂

题目描述
我们知道斐波那契数列的公式是：f ( n ) = f ( n − 1 ) + f ( n − 2 ) f(n) = f(n-1) + f(n-2)f(n)=f(n−1)+f(n−2)
其中f ( 1 ) = 1 f(1) = 1f(1)=1,f ( 2 ) = 1 f(2) = 1f(2)=1。

输入格式
输入一个正整数n nn( n ≤ 10 9 ) (n \le 10^9)(n≤10 
9
 )。

输出格式
输出f ( n ) m o d    ( 10 9 + 7 ) f(n) \mod (10^9 + 7)f(n)mod(10 
9
 +7)的值。

无法开1e9的一维数组！！使用矩阵幂进行计算。

代码：

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
#include <cstring>
#define MX 10005
using namespace std;
typedef vector<long long> vec;
typedef vector<vec> mat; 
int n;
//用二维vector表示矩阵
const int mod = 1e9+7;
//计算矩阵A*矩阵B
 mat mul(mat& a,mat& b)
 {
     mat c(a.size(),vec(b[0].size()));//矩阵a的行数，矩阵b的列数 
     for(int i = 0;i < a.size();i++)
     {
         for(int k = 0;k < b.size();k++)
         {
             for(int j = 0;j < b[0].size();j++)
             {
                 c[i][j] = (c[i][j] + a[i][k] * b[k][j]%mod)%mod;
             }
         }
     }
     return c;
 }
//计算a*b
mat pow(mat a,long long b)
{
    mat c(a.size(),vec(a.size()));
    for(int i = 0;i < a.size();i++)
    {
        c[i][i] = 1;
    }
    while(b > 0)
    {
        if(b & 1) c = mul(c,a);
        a = mul(a,a);
        b = b>>1;
     } 
     return c;
 } 
int main() {
    cin>>n;
    if(n == 0) cout<<0<<endl;
    if(n == 1 || n == 2) cout<<1<<endl;
    else{
        mat a(2,vec(2));
        a[0][0] = 1;
        a[0][1] = 1;
        a[1][0] = 1;
        a[1][1] = 0;
        mat res = pow(a,n-1);
        cout<<res[0][0] % mod<<endl;
    }
    return 0;
}