您的位置:首页 > 编程语言 > C语言/C++

HDU 1575 Tr A(矩阵快速幂)

2014-09-17 11:18 429 查看
题目地址:HDU 1575

矩阵快速幂裸题。

初学矩阵快速幂。以前学过快速幂,今天一看矩阵快速幂,原来其原理是一样的,这就好办多了。都是利用二分的思想不断的乘。只不过把数字变成了矩阵而已。

代码如下:

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>

using namespace std;
const int mod=9973;
int n;
struct matrix
{
int ma[20][20];
}init, res;
matrix Mult(matrix x, matrix y)
{
matrix tmp;
int i, j, k;
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
tmp.ma[i][j]=0;
for(k=0;k<n;k++)
{
tmp.ma[i][j]=(tmp.ma[i][j]+x.ma[i][k]*y.ma[k][j])%mod;
}
}
}
return tmp;
}
matrix Pow(matrix x, int k)
{
matrix tmp;
int i, j;
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
tmp.ma[i][j]=(i==j);
}
}
while(k)
{
if(k&1) tmp=Mult(tmp,x);
x=Mult(x,x);
k>>=1;
}
return tmp;
}
int main()
{
int t, i, j, ans, k;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&k);
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
scanf("%d",&init.ma[i][j]);
}
}
res=Pow(init,k);
ans=0;
for(i=0;i<n;i++)
{
ans=(ans+res.ma[i][i])%mod;
}
printf("%d\n",ans);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息