洛谷p3390矩阵快速幂普及/提高- 数论，矩阵快速幂

最新推荐文章于 2024-08-22 15:30:41 发布

chang_sheng1

最新推荐文章于 2024-08-22 15:30:41 发布

阅读量117

点赞数

分类专栏：洛谷

本文链接：https://blog.csdn.net/chang_sheng1/article/details/114708210

版权

洛谷专栏收录该内容

26 篇文章 0 订阅

订阅专栏

#include<iostream>
#include<cstdio>
using namespace std;
const int mod=1e9+7;
struct matrix{
    long long int m[101][101];
}A,I;
long long int n,k;
matrix operator*(const matrix&x,const matrix &y)
{
    matrix a;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            a.m[i][j]=0;
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            for(int k=1;k<=n;k++)
            {
                a.m[i][j]+=x.m[i][k]*y.m[k][j]%mod;
                a.m[i][j]%=mod;
            }
        }
    }
    return a;
}
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>A.m[i][j];
        }
    }
    for(int i=1;i<=n;i++)
    {
        I.m[i][i]=1;
    }
    while(k)
    {
        if(k%2==1) I=I*A;
        A=A*A;
        k=k>>1;
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(j!=1)
                cout<<" ";
            cout<<I.m[i][j];
        }
        cout<<endl;
    }
    return 0;
}