UVA 11551 - Experienced Endeavour(构造矩阵-水题)

题意:求一列序列的经过r次变化后的新序列,这些变化都是旧序列的某些已给位置的和产生新的项

思路:好水,直接构造01矩阵

//Accepted 45 ms C++ 4.8.2 1442
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;


const int mod= 1000;
int num[55];
int res[55];
int n,c;
struct mat
{
    int a[55][55];
    mat()
    {
        memset(a,0,sizeof(a));
    }
};
mat I;

mat mul(mat m1,mat m2)
{
    mat ans;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(m1.a[i][j])
                for(int k=1;k<=n;k++)
                    ans.a[i][k]=(ans.a[i][k]+m1.a[i][j]*m2.a[j][k])%mod;
    return ans;
}

mat quickmul(mat m,int k)
{
    mat ans;
    for(int i=1;i<=n;i++) ans.a[i][i]=1;
    while(k)
    {
        if(k&1) ans=mul(ans,m);
        m=mul(m,m);
        k>>=1;
    }
    return ans;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(I.a,0,sizeof(I.a));
        memset(res,0,sizeof(res));
        scanf("%d%d",&n,&c);
        for(int i=1;i<=n;i++) scanf("%d",&num[i]);
        for(int i=1;i<=n;i++) num[i]%=mod;
        for(int i=1;i<=n;i++)
        {
            int t,tmp;
            scanf("%d",&t);
            for(int j=1;j<=t;j++)
            {
                scanf("%d",&tmp);
                tmp++;
                I.a[i][tmp]=1;
            }
        }
        mat ans=quickmul(I,c);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                res[i]=(res[i]+num[j]*ans.a[i][j])%mod;
        for(int i=1;i<=n;i++) printf("%d%c",res[i],i==n?'\n':' ');

    }
    return 0;
}


你可能感兴趣的:(UVA 11551 - Experienced Endeavour(构造矩阵-水题))