POJ 3233 Matrix Power Series 动态规划(矩阵的幂)

一、题目大意

给出一个矩阵A,

输出矩阵B的每一项对M取余数的值。

二、解题思路

以二维矩阵为例,首先计算K=2的情况,我们设结果矩阵为B

有如下表达式

那么不难看出,需要的矩阵其实就是以下的两个矩阵相乘后的左上角的N*N个

然后我们再来考虑K=3的情况,我们设结果矩阵为C

我们来考虑如何把C表示成矩阵B和A相乘的状态。

不难看出C矩阵就是以下两个矩阵相乘后的N*N的左上角

也是如下矩阵的左上角

综上,我们发现计算B和C时,乘号右边的矩阵是相同的,只是我们需要保证前N排的后N列必须始终为A[0][0]..A[0][N-1],A[1][0]..A[1][N-1],...

保留那些值我们可以通过给右边的矩阵右下角放置零矩阵即可。

那么最终题目的答案可以表示成下的表达式。

三、代码

cpp 复制代码
#include <iostream>
#include <vector>
using namespace std;
typedef vector<int> vec;
typedef vector<vec> mat;
int M;
mat mul(mat &A, mat &B)
{
    mat C = mat(A.size(), vec(B[0].size()));
    for (int i = 0; i < A.size(); i++)
    {
        for (int j = 0; j < B[0].size(); j++)
        {
            for (int k = 0; k < B.size(); k++)
            {
                C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % M;
            }
        }
    }
    return C;
}
mat pow(mat &A, int N)
{
    mat B = mat(A.size(), vec(A[0].size()));
    for (int i = 0; i < B.size(); i++)
    {
        B[i][i] = 1;
    }
    while (N > 0)
    {
        if (N & 1)
        {
            B = mul(B, A);
        }
        A = mul(A, A);
        N >>= 1;
    }
    return B;
}
void solve()
{
    int N, K;
    scanf("%d%d%d", &N, &K, &M);
    mat A = mat(2 * N, vec(2 * N));
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            scanf("%d", &A[i][j]);
        }
    }
    for (int i = 0; i < N; i++)
    {
        A[i + N][i] = 1;
        A[i + N][i + N] = 1;
    }
    mat B = mat(2 * N, vec(2 * N));
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            B[i][j] = A[i][j];
            B[i][j + N] = A[i][j];
        }
    }
    A = pow(A, K - 1);
    A = mul(B, A);
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            printf("%d%c", A[i][j] % M, j + 1 == N ? '\n' : ' ');
        }
    }
}
int main()
{
    solve();
    return 0;
}
相关推荐
天天爱吃肉821820 分钟前
第十篇:电源设计的“能量矩阵”——无线充电与碳化硅LLC谐振
笔记·矩阵·汽车
Paddi93023 分钟前
Codeforces Round 1004 (Div. 1) C. Bitwise Slides
c++·算法
Luis Li 的猫猫1 小时前
深度学习中的知识蒸馏
人工智能·经验分享·深度学习·学习·算法
查理零世1 小时前
【蓝桥杯集训·每日一题2025】 AcWing 6118. 蛋糕游戏 python
python·算法·蓝桥杯
敲代码的小王!1 小时前
MD5加密算法和BCrypt密码加密算法
java·算法·安全
带娃的IT创业者2 小时前
机器学习实战(6):支持向量机(SVM)——强大的非线性分类器
算法·机器学习·支持向量机
孑么5 小时前
力扣 买卖股票的最佳时机
算法·leetcode·职场和发展·贪心算法·动态规划
诚信爱国敬业友善6 小时前
常见排序方法的总结归类
开发语言·python·算法
gentle_ice6 小时前
跳跃游戏 II - 贪心算法解法
数据结构·算法·leetcode·贪心算法
不只会拍照的程序猿7 小时前
从插入排序到希尔排序
java·开发语言·数据结构·算法·排序算法