洛谷P1962 斐波那契数列【矩阵快速幂模板】

题目描述：原题链接

求斐波那契数列第n项模1e9+7。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <map>
#include <vector>
using namespace std;
typedef long long ll;
typedef pair<int,string> PII;
const int N=10,mod=1e9+7;

ll n;
ll tmp[N][N],a[N][N],res[N][N];
void mul(ll a[][N],ll b[][N])
{
    memset(tmp,0,sizeof tmp);
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            for(int k=0;k<2;k++)
            tmp[i][j]=(tmp[i][j]+a[i][k]*b[k][j])%mod;
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
        a[i][j]=tmp[i][j];
}

int main()
{
    cin>>n;
    a[0][0]=1,a[0][1]=1,a[1][0]=1,a[1][1]=0;
    for(int i=0;i<2;i++) res[i][i]=1;
    while(n)
    {
        if(n&1) mul(res,a);
        mul(a,a);
        n>>=1;
    }
    cout<<res[0][1]<<endl;
    return 0;
}