Tr A (矩阵快速幂(模板))

【题目来源】https://vjudge.net/problem/HDU-1575
【题意】
中问题意不在解释。
【思路】
很水的一个快速幂题,而且又是给出了一个矩阵,所以想到矩阵快速幂。。
然而,并没有在敲完代码后直接过,因为在代码里我用到了两个矩阵:ans和base,他们俩的含义就和快速幂的一样,但是呢,本题是T组输入,然后我把他们俩用结构体定义在了外面,base是一直在更新(因为是把矩阵直接输入的),但是ans没有清0,所以又浪费了一点时间,,
【代码】

#include<set>
#include<map>
#include<stack>
#include<cmath>
#include<queue>
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<limits.h>
#include<algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int mod=9973;
typedef unsigned long long ll;
typedef long long LL;
int n,k;
struct mat
{
    int a[14][14];
    mat()
    {
        mem(a,0);
        for(int i=1; i<=10; i++)
            a[i][i]=1;
    }
} base;
mat operator*(mat s,mat t)
{
    mat r;
    mem(r.a,0);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            for(int p=1; p<=n; p++)
            {
                r.a[i][j]=r.a[i][j]+s.a[i][p]*t.a[p][j];
                if(r.a[i][j]>=mod)
                    r.a[i][j]%=mod;
            }
    return r;
}
void pow_mat(mat &ans)
{
    while(k)
    {
        if(k&1)
            ans=ans*base;
        base=base*base;
        k>>=1;
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        mat ans;
        scanf("%d%d",&n,&k);
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
                scanf("%d",&base.a[i][j]);
        pow_mat(ans);
        int sum=0;
        for(int i=1; i<=n; i++)
        {
            sum+=ans.a[i][i];
            if(sum>=mod)
                sum%=mod;
        }
        printf("%d\n",sum);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值