经典题目2 给定矩阵A,请快速计算出A^n(n个A相乘)的结果,输出的每个数都mod p。 由于矩阵乘法具有结合律,因此A^4 = A * A * A * A = (A*A) * (A*A) = A^2 * A^2。我们可以得到这样的结论:当n为偶数时,A^n = A^(n/2) * A^(n/2);当n为奇数时,A^n = A^(n/2) * A^(n/2) * A (其中n/2取整)。
这就告诉我们,计算A^n也可以使用二分快速求幂的方法。例如,为了算出A^25的值,我们只需要递归地计算出A^12、A^6、A^3的值即可。根据这里的一些结果,我们可以在计算过程中不断取模,避免高精度运算
<code>
#include<iostream>
using namespace std;
const int size=15;
struct arr
{
int map[size][size];
}init,unit;
int m,n;
arr mult(arr a,arr b)
{
int i,j,k;
arr c;
for (i=1;i<=n;i++)
{
for (j=1;j<=n;j++)
{
c.map[i][j] = 0;
for (k=1;k<=n;k++)
c.map[i][j]+=a.map[i][k]*b.map[k][j];
c.map[i][j]%=9973;
}
}
return c;
}
int getsum()
{
arr p,q,c;
int i,sum;
p=unit,q=init;
while(m!=1)
{
if(m%2==1)
{
m--;
p=mult(p,q);
}
else
{
m/=2;
q=mult(q,q);
}
}
c=mult(p,q);
sum=0;
for(i=1;i<=n;i++)
{
sum+=c.map[i][i];
sum%=9973;
}
return sum;
}
int main()
{
int T,i,j,k;
cin>>T;
while(T--)
{
cin>>n>>m;
for(i=1;i<=n;i++)//n*n·½Õó
{
for(j=1;j<=n;j++)
{
cin>>init.map[i][j];
if(i==j)unit.map[i][j]=1;
else unit.map[i][j]=0;
}
}
cout<<getsum()<<endl;
}return 0;
}
1万+

被折叠的 条评论
为什么被折叠?



