矩阵快速幂

矩阵乘法:
  行列式的概念:矩阵A=n*m,则称n为行,m为列。
  矩阵乘法的要求:A*B时:A=n*k,B=k*m。A的列等于B的行。新矩阵=n*m。
  新矩阵第n行,第m列的元素=A的第n行元素*B的第m列元素的乘积的和。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int mod=1e9+7;
struct mat{
    ll a[2][2];
};
mat matc(mat x,mat y){
    mat res;
    memset(res.a,0,sizeof(res.a));//important
    for(int i=0;i<2;i++)
       for(int j=0;j<2;j++)
          for(int k=0;k<2;k++)
               res.a[i][j]=(res.a[i][j]+x.a[i][k]*y.a[k][j])%mod;
    return res;           
}
inline int pow(int n){
    mat c,res;
    memset(res.a,0,sizeof(res.a));
    c.a[0][0]=1; c.a[0][1]=1;
    c.a[1][0]=1; c.a[1][1]=0;
    for(int i=0;i<n;i++) res.a[i][i]=1;//单位矩阵 
    while(n){
        if(n&1) res=matc(res,c);
        c=matc(c,c);
        n>>=1;
    }
    return res.a[0][1];
}
int main(){
    int n;
    scanf("%d",&n);
    cout<<pow(n);
}

 

posted @ 2017-10-23 10:23  TimDucan  阅读(2847)  评论(0编辑  收藏  举报