UVA-11551 Experienced Endeavour

给出初始数组和操作，每次操作后，数组中每个数字将变为原来数组中若干个数字之和

问r次操作后数组是什么样子

可以把操作转化为矩阵乘法，然后用矩阵快速幂求解

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const ll MOD=1000;
int N;
struct node
{
	ll a[55][55];
};
node shu,ans,mp;
//shu是输入的矩阵，ans是所求答案
node matrix(node x,node y)
{
    for(int i=1;i<=N;i++)
        for(int j=1;j<=N;j++){
            mp.a[i][j]=0;
            for(int p=1;p<=N;p++)
                mp.a[i][j]=(mp.a[i][j]+x.a[i][p]*y.a[p][j]+MOD)%MOD;
            //矩阵乘法 
        }
    return mp;
}
ll a[55];
void work(ll k)
{//矩阵快速幂 
	for(int i=1;i<=N;i++)
			for(int j=1;j<=N;j++)
				ans.a[i][j]=0;
	for(int i=1;i<=N;i++) ans.a[i][i]=1;
	node t=shu;
    while(k){
        if(k&1) 
            ans=matrix(ans,t);
        k>>=1;
        t=matrix(t,t);
    }
}
int main()
{
	int T,n,m;
	scanf("%d",&T);
	while(T--)
	{
		int x,b;
		memset(shu.a,0,sizeof(shu.a));
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
			scanf("%lld",&a[i]);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&x);
			while(x--)
			{
				scanf("%d",&b);
				shu.a[i][b+1]=1;
			}
		}
		N=n;
		work(m);
		for(int i=1;i<=n;i++) 
		{
			ll ret=0;
			for(int j=1;j<=n;j++)
				ret=(ret+ans.a[i][j]*a[j])%MOD;
			printf("%lld",ret);
			if(i!=n) printf(" ");
		}
		printf("\n");
	}
	return 0;
}