首页 > 其他 > 详细

又见矩阵快速幂 模板贴起来

时间:2015-03-03 16:37:23      阅读:160      评论:0      收藏:0      [点我收藏+]
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <queue>
#include <cmath>
#include <stack>
#include <map>

#pragma comment(linker, "/STACK:1024000000")
#define EPS (1e-8)
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f

using namespace std;

const int MAXN = 61;

struct MAT
{
    int row,col;
    int mat[MAXN][MAXN];

    void Init(int R,int C,int val)
    {
        row = R,col = C;

        for(int i = 1;i <= row; ++i)
            for(int j = 1;j <= col; ++j)
                    mat[i][j] = (i == j ? val : 0);
    }

    MAT Multi(MAT c,int MOD)
    {
        MAT tmp;
        tmp.Init(this->row,c.col,0);

        int i,j,k;

        for(i = 1;i <= tmp.row; ++i)
            for(j = 1;j <= tmp.col; ++j)
                for(k = 1;k <= this->col; ++k)
                    (tmp.mat[i][j] += this->mat[i][k]*c.mat[k][j]) %= MOD;
        return tmp;
    }

    MAT Quick(int n,int MOD)
    {
        MAT res,tmp = *this;

        res.Init(row,col,1);

        while(n)
        {
            if(n&1)
                res = res.Multi(tmp,MOD);
            tmp = tmp.Multi(tmp,MOD);
            n >>= 1;
        }

        return res;
    }

    void Output()
    {
        cout<<"         ****************        "<<endl;
        int i,j;
        for(i = 1;i <= row; ++i)
        {
            for(j = 1;j <= col; ++j)
                    printf("%3d ",mat[i][j]);
            puts("");
        }
        cout<<"         &&&&&&&&&&&&&       "<<endl;

    }

};

int main()
{
    freopen("data1.in","r",stdin);
    int T;

    scanf("%d",&T);

    MAT A;

    int n,m,i,j,sum;

    while(T--)
    {
        scanf("%d %d",&n,&m);

        A.Init(n,n,0);

        for(i = 1;i <= n; ++i)
            for(j = 1;j <= n; ++j)
                    scanf("%d",&A.mat[i][j]);

        A = A.Quick(m,9973);

        for(sum = 0 ,i = 1;i <= n; ++i)
           (sum += A.mat[i][i] ) %= 9973;

        printf("%d\n",sum);
    }
    return 0;
}

又见矩阵快速幂 模板贴起来

原文:http://blog.csdn.net/zmx354/article/details/44039655

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!