码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2019-07-30 21:41:58      阅读:94      评论:0      收藏:0      [点我收藏+]

标签:multi   快速   复制   code   初始   mem   while   大小   斐波那契   

注意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

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

标签:multi   快速   复制   code   初始   mem   while   大小   斐波那契   

原文地址:https://www.cnblogs.com/Diliiiii/p/11272674.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!