51nod 矩阵快速幂模版题

给出一个N * N的矩阵,其中的元素均为正整数。求这个矩阵的M次方。由于M次方的计算结果太大,只需要输出每个元素Mod (10^9 + 7)的结果。
Input
第1行:2个数N和M,中间用空格分隔。N为矩阵的大小,M为M次方。(2 <= N <= 100, 1 <= M <= 10^9)
第2 - N + 1行:每行N个数,对应N * N矩阵中的1行。(0 <= N[i] <= 10^9)
Output
共N行,每行N个数,对应M次方Mod (10^9 + 7)的结果。
Input示例
2 3
1 1
1 1
Output示例
4 4
4 4

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
#include <set>
#include <algorithm>
typedef long long ll;
using namespace std;
const int N=100+1;
const ll mod=1e9+7;
int n;
struct node
{
    ll cnt[N][N];
    void CSH()
    {
        memset(cnt,0,sizeof(cnt));
    }
    void init()
    {
        CSH();
        for(int i=0;i<n;i++)
            cnt[i][i]=1;
    }
};
node mul(node x,node y)
{
    node b;
    b.CSH();
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            if(x.cnt[i][j])
                for(int k=0;k<n;k++)
                    b.cnt[i][k]=(x.cnt[i][j]*y.cnt[j][k]+b.cnt[i][k])%mod;
    return b;
}
node quick(node x,ll m)
{
    node sum;
    sum.init();
    while(m)
    {
        if(m%2) sum=mul(sum,x);
        x=mul(x,x);
        m/=2;
    }
    return sum;
}
int main()
{
    ll m;
    node a;
    scanf("%d%lld",&n,&m);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            scanf("%lld",&a.cnt[i][j]);

    node k=quick(a,m);
    for(int i=0;i<n;i++)
    {
        printf("%lld",k.cnt[i][0]);
        for(int j=1;j<n;j++)
            printf(" %lld",k.cnt[i][j]);
        puts("");
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值