码迷,mamicode.com
首页 > 其他好文 > 详细

HDU1575:Tr A(矩阵快速幂模板题)

时间:2014-11-08 16:34:23      阅读:142      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   io   color   ar   os   sp   for   

 
#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <cstdio>
#include <algorithm>
#define mod 9973
using namespace std;
struct matrix
{
    int a[11][11];
} init,res;
int n,k;
matrix Mult(matrix x,matrix y)
{
    matrix tmp;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            tmp.a[i][j]=0;
            for(int k=0; k<n; k++)
                tmp.a[i][j]=(tmp.a[i][j]+x.a[i][k]*y.a[k][j])%mod;
        }
    }
    return tmp;
}
matrix Pow(matrix x,int k)
{
    matrix tmp;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
            tmp.a[i][j]=(i==j);
    }
    while(k)
    {
        if(k&1)
            tmp=Mult(tmp,x);
        k>>=1;
        x=Mult(x,x);
    }
    return tmp;
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n>>k;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
                scanf("%d",&init.a[i][j]);
        }
        res=Pow(init,k);
        int sum=0;
        for(int i=0; i<n; i++)
            sum=(sum+res.a[i][i])%mod;
        cout<<sum<<endl;
    }
    return 0;
}

 

HDU1575:Tr A(矩阵快速幂模板题)

标签:style   blog   http   io   color   ar   os   sp   for   

原文地址:http://www.cnblogs.com/zhangmingcheng/p/4083426.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!