HDU1575Tr A（矩阵相乘与快速幂）

Tr A  hdu1575

 

就是一个快速幂的应用：

只要知道怎么求矩阵相乘！！（比赛就知道会超时，就是没想到快速幂！！！）

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
__int64 a[15][15],b[15][15],c[15][15];
int n;
int main()
{
    int t,i,j,m,k,d;
    __int64 sum;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
            {scanf("%I64d",&a[i][j]);b[i][j]=a[i][j];}
        d=0;
        while(m)
        {
            if(m%2&&d==0)//因为快速幂规定起始为1，则第一次就来要分开考虑！！
            {
                for(i=0;i<n;i++)
                    for(j=0;j<n;j++)
                        a[i][j]=b[i][j];
                d=1;
            }
            else if(m%2)
            {
                memset(c,0,sizeof(c));
                for(i=0;i<n;i++)
                    for(j=0;j<n;j++)
                    for(k=0;k<n;k++)
                        c[i][j]+=a[i][k]*b[k][j]%9973;
                for(i=0;i<n;i++)
                    for(j=0;j<n;j++)
                        a[i][j]=c[i][j];
            }
            m=m/2;
            memset(c,0,sizeof(c));
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
        for(k=0;k<n;k++)
            c[i][j]+=b[i][k]*b[k][j]%9973;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            b[i][j]=c[i][j];
        }
        sum=0;
        for(i=0;i<n;i++)
            sum+=a[i][i];
        printf("%I64d\n",sum%9973);
    }
    return 0;
}

 

附上快速幂的模板：

__int64 power(__int64 p,__int64 n)  //反复平方法求(p^n)%mod
{
    __int64 sq=1;
    while(n>0)
    {
        if(n%2)
            sq=(sq*p)%mod;
        n/=2;
        p=p*p%mod;
    }
    return sq;
}