矩陣快速冪

N=2//矩陣大小
ll quick(ll a,ll b){//快速冪
    ll ans=1;
    while(b){
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b/=2;
    }
    return ans%mod;
}


struct sa
{
    ll m[N][N];
};
sa Mul(sa a,sa b)//矩陣乘法
{
    sa c;
    memset(c.m,0, sizeof(c.m));
    for(int i=0;i<N;i++)
        for(int j=0;j<N;j++)
            for(int k=0;k<N;k++)
                c.m[i][j]=(c.m[i][j]+a.m[i][k]*b.m[k][j]%mod)%mod;
    return c;
}

sa quickjz(sa a,ll n)//矩陣快速冪
{
    sa res;
    memset(res.m,0, sizeof(res.m));
    //res.m[0][0]=1;
    //res.m[1][1]=1;
    for(int i=0;i<N;i++)res.m[i][i]=1;
    while(n)
    {
        if(n&1)
            res=Mul(res,a);
        n/=2;
        a=Mul(a,a);
    }
    return res;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章