矩阵快速幂模板

struct M
{
    LL a[3][3];
    void init1()//化单位矩阵
    {
        memset(a, 0, sizeof(a));
        a[0][0]=1;
        a[1][1]=1;
        a[2][2]=1;
    }
    void init0()//清空矩阵
    {
       memset(a, 0, sizeof(a));
    }
};
M mul (M a,M b)
{
    M ans;
    for (int i=0; i<maxn; i++)
    {
        for (int j=0; j<maxn; j++)
        {
            ans.a[i][j]=0;
            for (int k=0; k<maxx; k++)
            {
                ans.a[i][j]+=a.a[i][k]*b.a[k][j];
                ans.a[i][j]%=mod;
            }
        }
    }
    return ans;
}
M qpow(M a,LL n)//快速幂部分
{
    M ans;
    ans.init();
    //这里也可以初始化我要b*a^n中的b矩阵
    while(n)
    {
        if (n&1)ans=mul(ans,a);
        a=mul(a,a);
        n/=2;
    }
    return ans;
}
void output(M a)//随时打印判断
{
    for(int i=0; i<3; ++i)
    {
        for(int j=0; j<3; ++j)
        {
            cout << a.a[i][j] << " ";
        }
        cout << endl;
    }
}

 

全部评论

相关推荐

字节一直是我的白月光,考虑到转正还是拒了日常实习。
从明天开始狠狠卷JV...:为什么你释放的offer没流到我头上
点赞 评论 收藏
分享
07-02 13:52
武汉大学 golang
点赞 评论 收藏
分享
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
昨天 17:30
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务