矩阵快速幂模板

矩阵快速幂模板

复制代码
#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;
typedef long long ll;
const int lg= 1e9+7;
const int maxn = 200;
struct node {
    ll m[maxn][maxn];
}ans,res;

node Mul(node a,node b,ll n)
{
    node tmp;
    memset(tmp.m,0,sizeof(tmp));
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
        {
            for(int k=1; k<=n; k++)
                tmp.m[i][j] = (tmp.m[i][j]+a.m[i][k]*b.m[k][j]%lg)%lg;
        }
    }
    return tmp;
}
void jzksm(ll n,ll k)
{
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
        {
            if(i==j)ans.m[i][j] = 1;
            else ans.m[i][j] = 0;
        }
    }
    while(k)
    {
        if(k&1)ans = Mul(ans,res,n);
        res = Mul(res,res,n);
        k>>=1;
    }
}
int main(){
    ll n,k;
    scanf("%lld%lld",&n,&k);
    for(int i=1; i<=n; i++)
    {
        for(int j=1;j<=n;j++)
            scanf("%lld",&res.m[i][j]);
    }
    jzksm(n,k);
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n; j++)
            printf("%lld%c",ans.m[i][j]," \n"[j==n]);
    }
    return 0;
}
 
复制代码

 

posted @   ckxkexing  阅读(144)  评论(0编辑  收藏  举报
编辑推荐:
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
· .NET 9 new features-C#13新的锁类型和语义
阅读排行:
· Sdcb Chats 技术博客:数据库 ID 选型的曲折之路 - 从 Guid 到自增 ID,再到
· 语音处理 开源项目 EchoSharp
· 《HelloGitHub》第 106 期
· Spring AI + Ollama 实现 deepseek-r1 的API服务和调用
· 使用 Dify + LLM 构建精确任务处理应用
点击右上角即可分享
微信分享提示