快速幂与矩阵快速幂-------------------Tr A

先看快速幂:
这里转载一篇大牛的快速幂文章,写的很详细,注意后面的计算利用了公式:
(a^b) mod c = (a mod c)^b mod c 注意这里的mod c是对最后整体的取余
https://blog.csdn.net/qq_19782019/article/details/85621386
然后再看矩阵快速幂:
这里还是转载一篇大牛的文章,https://blog.csdn.net/red_red_red/article/details/90208713?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-8.channel_param&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-8.channel_param
矩阵快速幂是分成矩阵乘法和快速幂计算两部分
这个题一个模板题,对于一般的快速幂算法还能勉强理解,对于矩阵快速幂以我的理解力确实整不来,只能记下模板,这里是我找到的差不多最标准的模板例题了:(https://blog.csdn.net/forever_kirito/article/details/77141006)

#include
#include
#include
using namespace std;
const int mod = 9973;
int k,n;
int a[20][20],temp[20][20],res[20][20];
void Mul(int a[][20],int b[][20])///矩阵乘法
{
    memset(temp,0,sizeof(temp));
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            for(int k=0;k<n;k++)
                temp[i][j]=(temp[i][j]+a[i][k]*b[k][j])%mod;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            a[i][j]=temp[i][j];
}
void fun(int a[][20],int x)///快速幂
{
    memset(res,0,sizeof(res));
    for(int i=0;i<n;i++)
        res[i][i]=1;
    while(x){
        if(x&1){
            Mul(res,a);
        }
        Mul(a,a);
        x>>=1;
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&k);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&a[i][j]);
        fun(a,k);
        int ans=0;
        for(int i=0;i<n;i++)
            ans+=res[i][i]%mod;
        printf("%d\n",ans%mod);
    }
    return 0;
}

你可能感兴趣的:(快速幂,算法)