Dili_iiii

编码全要靠底力

【模板】【矩阵快速幂】求第n项斐波那契

注意a数组初始化,调用Pow(a,n,w),n是第几项,求斐波那契中w默认为2

ll tmp[2][2],res[2][2];
void multi(ll a[][2],ll b[][2],int n)
{
    memset(tmp,0,sizeof(tmp));
    for(ll i=0;i<n;i++)
    {
       for(ll j=0;j<n;j++)
       {
        for(ll k=0;k<n;k++)
        {
        tmp[i][j]+=(a[i][k]*b[k][j])%p;
        }
        tmp[i][j]=tmp[i][j]%p;
      }
   }
    for(ll i=0;i<n;i++)
        for(ll j=0;j<n;j++)
        a[i][j]=tmp[i][j];
}
void Pow(ll a[][2],ll m,int n)
{
    memset(res,0,sizeof(res));//m是幂,n是矩阵大小
    for(ll i=0;i<n;i++) res[i][i]=1;
    while(m)
    {
        if(m&1)
        multi(res,a,n);//res=res*a;复制直接在multi里面实现了;
        multi(a,a,n);//a=a*a
        m>>=1;
    }
}
ll qm(ll a,ll b)
{
    ll ret=1;
    while(b)
    {
        if(b&1)
        {
            ret=ret*a%p;
        }
        a=a*a%p;
        b>>=1;
    }
    return ret;
}
//Pow(a,n,2) 调用  初始化:a[0][0]=a[0][1]=a[1][0]=1,a[1][1]=0
posted @ 2019-07-30 20:48  Dili_iiii  阅读(216)  评论(0编辑  收藏  举报