hdu 1575 try a 矩阵快速幂

#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<queue>
#include<stack>
#include<algorithm>
#include<iostream>
using namespace std;
#define ll long long int
const int m=9973;

ll a[20][20],k;
int n;
void pow_mod()
{
    int i,j,p,q;
    ll t[20][20],temp[20][20];
    q=k;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            {
                t[i][j]=a[i][j];
                if(i==j) a[i][j]=1;
                else a[i][j]=0;
            }

    while(q)
    {
        if(q%2==1)
        {
            for(i=0;i<n;i++)
                for(j=0;j<n;j++)
                {
                    temp[i][j]=a[i][j];
                    a[i][j]=0;
                }
            for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                for(k=0;k<n;k++)
                    a[i][j]=(a[i][j]+temp[i][k]*t[k][j])%m;
        }
        for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            {
                temp[i][j]=t[i][j];
                t[i][j]=0;
            }
            for(i=0;i<n;i++)
                for(j=0;j<n;j++)
                    for(k=0;k<n;k++)
                     t[i][j]=(t[i][j]+temp[i][k]*temp[k][j])%m;
        q/=2;
    }
}
int main()
{
    int _,i,j;
    ll ans;
    scanf("%d",&_);
    while(_--)
    {
        scanf("%d%lld",&n,&k);

        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                scanf("%lld",&a[i][j]);
        pow_mod();
        for(i=0,ans=0;i<n;i++)
        {
            ans=(ans+a[i][i])%m;
        }
        printf("%lld\n",ans);
    }
    return 0;
}

你可能感兴趣的:(hdu 1575 try a 矩阵快速幂)