您的位置:首页 > 其它

数学专项matrix:UVa 11551

2013-08-07 23:21 253 查看
快速幂模板题。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=60;
const int mod=1000;
typedef int matrix[maxn][maxn];
int a[maxn],b[maxn];
int n,r;
void mat_mul(matrix A,matrix B,matrix res)
{
matrix C;
memset(C,0,sizeof(C));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
C[i][j]=(C[i][j]+A[i][k]*B[k][j])%mod;
memcpy(res,C,sizeof(C));
}
void mat_pow(matrix A,int m,matrix res)
{
matrix a,r;
memset(r,0,sizeof(r));
memcpy(a,A,sizeof(a));
for(int i=0;i<n;i++) r[i][i]=1;
while(m)
{
if(m&1) mat_mul(r,a,r);
m>>=1;
mat_mul(a,a,a);
}
memcpy(res,r,sizeof(r));
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&r);
for(int i=0;i<n;i++) scanf("%d",&a[i]);
matrix A;
memset(A,0,sizeof(A));
for(int i=0;i<n;i++)
{
int t;
scanf("%d",&t);
for(int j=0;j<t;j++)
{
int p;
scanf("%d",&p);
A[i][p]=1;
}
}
mat_pow(A,r,A);
memset(b,0,sizeof(b));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
b[i]=(b[i]+A[i][j]*a[j])%mod;
for(int i=0;i<n;i++)
if(i) printf(" %d",b[i]);
else printf("%d",b[i]);
puts("");
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: