[HDU1575]Tr A（矩阵乘法）

最新推荐文章于 2021-03-01 12:30:55 发布

原创最新推荐文章于 2021-03-01 12:30:55 发布 · 876 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#POJ #HDU #矩阵

题解同时被 2 个专栏收录

1097 篇文章

订阅专栏

矩阵

21 篇文章

订阅专栏

本文介绍了一种利用矩阵快速幂算法解决特定问题的方法。通过定义矩阵运算和快速幂操作，可以高效地计算矩阵的高次幂。适用于求解递推序列、线性代数等领域的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

传送门

题意
给出一个矩阵A和整数k，求 $A^k$

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int Mod=9973;
const int max_n=15;
int T,n,k,ans;
struct hp{int a[max_n][max_n];}st,m,matrix;

inline hp cheng(hp a,hp b)
{
    hp ans;
    memset(ans.a,0,sizeof(ans.a));
    for (int i=1;i<=n;++i)
        for (int j=1;j<=n;++j)
            for (int k=1;k<=n;++k)
                ans.a[i][j]=(ans.a[i][j]+a.a[i][k]*b.a[k][j]%Mod)%Mod;
    return ans;
}
inline hp matrix_fast_pow(hp a,int p)
{
    hp ans=st;
    for (;p;p>>=1,a=cheng(a,a))
        if (p&1)
            ans=cheng(ans,a);
    return ans;
}
int main()
{
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d%d",&n,&k);
        memset(st.a,0,sizeof(st.a));
        for (int i=1;i<=n;++i) st.a[i][i]=1;
        for (int i=1;i<=n;++i)
            for (int j=1;j<=n;++j)
                scanf("%d",&m.a[i][j]);
        matrix=matrix_fast_pow(m,k);
        ans=0;
        for (int i=1;i<=n;++i)
            ans=(ans+matrix.a[i][i])%Mod;
        printf("%d\n",ans);
    }
}