Experienced Endeavour UVA - 11551

Experienced Endeavour UVA - 11551
思路:水题,理解好题意可以直接构造 矩阵。给出n,r,n是代表接下来又n+1行,先给出一行n个数的数列ai,接下来n行开头给一个个数x,输入x个数bi,代表第ai接下来的变化要加上数列a(bi)的值,所以只要给构造的举证该位置置为1。r代表幂次。

#include <cstdio> 
#include<iostream> 
#include <cstring> 
//数列求和与矩阵求和
using namespace std;
const int mod=1000;
typedef long long ll;
struct  matrix{
    ll x[50][50];
};
ll n,r;
matrix multi(matrix a,matrix b){
    matrix temp;
    memset(temp.x,0,sizeof(temp.x));
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            for(int k=0;k<n;k++)
            {
                temp.x[i][j]+=a.x[i][k]*b.x[k][j];
                temp.x[i][j]%=mod;//负数取模的问题,除法取模
            }
    return temp;
}
matrix quick_multi(matrix a,ll n)//矩阵快速幂
{
    matrix temp=a;
    n--;
    while(n){
        if(n&1)
            temp=multi(temp,a);
        a=multi(a,a);
        n>>=1;
    }
    return temp;
}
int a[50];
matrix b;
int main()
{
    int t,x,temp;
    scanf("%d",&t);
    while(t--)
    {
        memset(b.x,0,sizeof(b.x));
        scanf("%lld%lld",&n,&r);
        for(int i=0;i<n;i++) scanf("%d",&a[i]);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&x);
            while(x--){
                scanf("%d",&temp);
                b.x[i][temp]=1;
            }
        }
        b=quick_multi(b,r);
        for(int i=0;i<n;i++)
        {   int ans=0;
            for(int j=0;j<n;j++)
            {
                ans+=b.x[i][j]*a[j];
            }
            i==0?printf("%d",ans%mod):printf(" %d",ans%mod);;
        }
        printf("\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Miranda_ymz/article/details/81835212