HDU 1575 Tr A

weixin_33795833

于 2015-07-17 14:24:00 发布

阅读量56

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zufezzt/p/4654372.html

本文介绍了一种使用矩阵快速幂算法解决特定数学问题的方法。通过定义矩阵运算符重载实现矩阵乘法，并利用快速幂思想计算矩阵的幂次，最终求解特定数值。适用于计算机科学中的算法竞赛题目。

矩阵快速幂

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

const int maxn=15;
const int MOD=9973;
int n,k;

struct Matrix
{
    int matrix[maxn][maxn];
    Matrix operator*(Matrix b);
};

Matrix Matrix::operator*(Matrix b)
{
    Matrix g;
    int i,j,k;
    for(i=1; i<=n; i++)
        for(j=1; j<=n; j++)
        {
            int sum=0;
            for(k=1; k<=n; k++)
                sum=(sum+(matrix[i][k]*b.matrix[k][j])%MOD)%MOD;
            g.matrix[i][j]=sum;
        }
    return g;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int i,j;
        scanf("%d%d",&n,&k);
        Matrix a;
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
                scanf("%d",&a.matrix[i][j]);
        Matrix c;
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                if(i==j) c.matrix[i][j]=1;
                else c.matrix[i][j]=0;
            }
        }
        while(k!=0)
        {
            if(k%2==1) c=c*a,k--;
            else a=a*a,k=k/2;
        }
        int ans=0;
        for(i=1; i<=n; i++)
            ans=(ans+c.matrix[i][i])%MOD;
        printf("%d\n",ans);
    }
    return 0;
}