hdu1575

矩阵快速幂的应用,先定义一个结构体,定义矩阵和矩阵乘法,方便以后直接调用,再写一个矩阵快速幂的函数,进行相关运算。其实对于比较大的指数运算时,快速幂的优势是很明显的,时间复杂度降低到很小。

#include 
#include 
#include 
#define mode 9973
using namespace std;

int n;

struct matrix
{
    int mat[11][11];
    matrix operator*(const matrix m) const
    {
        matrix tmp;
        for(int i = 0;i < n;i ++)
            for(int j = 0;j < n;j ++)
            {
                tmp.mat[i][j] = 0;
                for(int k = 0;k < n;k ++)
                {
                    tmp.mat[i][j] += mat[i][k] * m.mat[k][j] % mode;
                    tmp.mat[i][j] %= mode;
                }
            }
            return tmp;
    }
};

long long pow(matrix &m,int k)
{
    matrix ans;
    memset(ans.mat,0,sizeof(ans.mat));
    for(int i = 0;i < n;i ++)
        ans.mat[i][i] = 1;
    while(k)
    {
        if(k&1) ans = ans * m;
        k >>= 1;
        m = m * m;
    }
    long long sum = 0;
    for(int i = 0;i < n;i ++)
        {sum += ans.mat[i][i] % mode;
    sum %= mode;}
    return sum;
}

int main()
{
    //ifstream cin("in.txt");
    matrix cat;
    int T,k;cin >> T;
    while(T --)
    {
        cin >> n >> k;
        for(int i = 0;i < n;i ++)
            for(int j = 0;j < n;j ++)
            cin >> cat.mat[i][j];
        cout << pow(cat,k) << endl;
    }
    return 0;
}


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