洛谷P3390 【模板】矩阵快速幂

题目背景

矩阵快速幂

题目描述

给定n*n的矩阵A,求A^k

输入输出格式

输入格式:

 

第一行,n,k

第2至n+1行,每行n个数,第i+1行第j个数表示矩阵第i行第j列的元素

 

输出格式:

 

输出A^k

共n行,每行n个数,第i行第j个数表示矩阵第i行第j列的元素,每个元素模10^9+7

 

输入输出样例

输入样例#1:
2 1
1 1
1 1
输出样例#1:
1 1
1 1

说明

n<=100, k<=10^12, |矩阵元素|<=1000 算法:矩阵快速幂

题解:蒟蒻的模板库

代码:

复制代码
#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 1000000007
using namespace std;
int n;
struct matrix{
    long long m[101][101];
}A;
matrix mul(matrix a,matrix b){
    matrix t;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            t.m[i][j]=0;
            for(int k=1;k<=n;k++){
                t.m[i][j]=(t.m[i][j]+a.m[i][k]*b.m[k][j])%mod;
            }
        }
    }
    return t;
}
matrix fast_mul(matrix a,long long k){
    matrix ret=a,now=a;
    k--;
    while(k){
        if(k&1)ret=mul(ret,now);
        now=mul(now,now);
        k>>=1;
    }
    return ret;
}
long long k;
int main(){
    scanf("%d%lld",&n,&k);
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      scanf("%d",&A.m[i][j]);
    A=fast_mul(A,k);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
        printf("%lld ",A.m[i][j]);
        printf("\n");
    }
    return 0;
}
复制代码

 

posted @   ANhour  阅读(153)  评论(0编辑  收藏  举报
编辑推荐:
· .NET 9 new features-C#13新的锁类型和语义
· Linux系统下SQL Server数据库镜像配置全流程详解
· 现代计算机视觉入门之:什么是视频
· 你所不知道的 C/C++ 宏知识
· 聊一聊 操作系统蓝屏 c0000102 的故障分析
阅读排行:
· DeepSeek V3 两周使用总结
· 回顾我的软件开发经历(1)
· C#使用yield关键字提升迭代性能与效率
· 低成本高可用方案!Linux系统下SQL Server数据库镜像配置全流程详解
· 4. 使用sql查询excel内容
点击右上角即可分享
微信分享提示