POJ3233Matrix Power Series(矩阵快速幂)

 


题意

题目链接

给出n×n的矩阵A,求i=1kAi,每个元素对m取模

Sol

考虑直接分治

k为奇数时

i=1kAi=i=1k/2+1Ai+Ak/2+1(i=1k/2Ai)

k为偶数时

sumi=1k=i=1k/2Ai+Ak/2(i=1k/2Ai)

 

当然还可以按套路对前缀和构造矩阵也是可以做的。

复制代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<map>
#define LL long long 
using namespace std;
int N, K, mod;
int mul(int x, int y) {
    if(1ll * x * y > mod) return 1ll * x * y % mod;
    else return 1ll * x * y;
}
int add(int x, int y) {
    if(x + y > mod) return x + y - mod;
    else return x + y;
}
struct Matrix {
    int m[31][31];
    Matrix() {
        memset(m, 0, sizeof(m));
    }
    bool operator < (const Matrix &rhs) const {
        for(int i = 1; i <= N; i++)
            for(int j = 1; j <= N; j++)
                if(m[i][j] != rhs.m[i][j])
                    return m[i][j] < rhs.m[i][j];
        return 1;
    }
    Matrix operator * (const Matrix &rhs) const {
        Matrix ans;
        for(int k = 1; k <= N; k++)
            for(int i = 1; i <= N; i++)
                for(int j = 1; j <= N; j++)
                    ans.m[i][j] = add(ans.m[i][j], mul(m[i][k], rhs.m[k][j]));
        return ans;
    }
    Matrix operator + (const Matrix &rhs) const {
        Matrix ans;
        for(int i = 1; i <= N; i++)
            for(int j = 1; j <= N; j++)
                ans.m[i][j] = add(m[i][j], rhs.m[i][j]);
        return ans;
    }
}a;
Matrix getbase() {
    Matrix base;
    for(int i = 1; i <= N; i++) base.m[i][i] = 1;
    return base;
}
Matrix fp(Matrix a, int p) {
    Matrix base = getbase();
    while(p) {
        if(p & 1) base = base * a;
        a = a * a; p >>= 1;
    }
    return base;
}
Matrix solve(int k) {
    if(k == 1) return a;
    Matrix res = solve(k / 2);
    if(k & 1) {
        Matrix po = fp(a, k / 2 + 1);
        return res + po + po * res;
    }
    else return res + fp(a, k / 2) * res;

}
main() {
//    freopen("a.in", "r", stdin);
    cin >> N >> K >> mod;
    for(int i = 1; i <= N; i++)
        for(int j = 1; j <= N; j++)
            cin >> a.m[i][j];
    Matrix ans = solve(K);
    for(int i = 1; i <= N; i++, puts(""))
        for(int j = 1; j <= N; j++)
            printf("%d ", ans.m[i][j] % mod);
}
复制代码
posted @   自为风月马前卒  阅读(567)  评论(0编辑  收藏  举报
编辑推荐:
· Java 中堆内存和栈内存上的数据分布和特点
· 开发中对象命名的一点思考
· .NET Core内存结构体系(Windows环境)底层原理浅谈
· C# 深度学习:对抗生成网络(GAN)训练头像生成模型
· .NET 适配 HarmonyOS 进展
阅读排行:
· 手把手教你更优雅的享受 DeepSeek
· 腾讯元宝接入 DeepSeek R1 模型,支持深度思考 + 联网搜索,好用不卡机!
· AI工具推荐:领先的开源 AI 代码助手——Continue
· 探秘Transformer系列之(2)---总体架构
· V-Control:一个基于 .NET MAUI 的开箱即用的UI组件库

Contact with me

点击右上角即可分享
微信分享提示