矩阵快速幂(模板)

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int mod = 1e9 + 7;
ll n, k;
struct MUL
{
    ll m[105][105];
}res;
MUL mul(MUL a, MUL b)
{
    MUL tmp;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            tmp.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++)
            {
                tmp.m[i][j] = (tmp.m[i][j] % mod + (a.m[i][k] % mod * b.m[k][j] % mod) % mod) % mod;
            }
        }
    }
    return tmp;
}
MUL quickpow(MUL a, ll b, ll modd)
{
    if (b == 1)return a;
    else
    {
        if (b % 2 == 0)
        {
            return quickpow(mul(a, a), b / 2, modd);
        }
        else
        {
            return mul(quickpow(mul(a, a), b / 2, modd), a);
        }
    }
}
int main()
{
    MUL A;
    cin >> n >> k;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            cin >> A.m[i][j];
        }
    }
    MUL result = quickpow(A, k, mod);
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            cout << result.m[i][j] << " ";
        }
        cout << endl;
    }
}

 

全部评论

相关推荐

09-18 20:41
百度_Java
要个offer怎么这...:哈哈哈哈哈哈,我也拿了0x10000000个offer,秋招温啦啦啦,好开心
我的秋招日记
点赞 评论 收藏
分享
笑着秋招😊:我一直认为努力有回报是一件很幸福很幸福的事情,恭喜你
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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