codeforce612 E. Wet Shark and Blocks 矩阵快速幂加速DP

codeforce612 E. Wet Shark and Blocks

有b个块，每个块中有相同的n个元素，在每个块中取1个元素，使这b个数所组成的b位整数%x==k，求此方法数

在第一个块中取a,a'=a%x，第二个块中一个元素与a'所组成的数为b=a'*10+b,b'=b%x

以此类推，最终可得到模最后几位得到k的状态

转移方程：类似邻接矩阵的n次方，某个点的值表示达到这个点的方案数

#include<bits/stdc++.h>
using namespace std;
const int MAX=5e4+5;
const long long MOD=1e9+7;
int n,b,K,x,a[MAX];
struct Mat
{
    long long m[101][101];
    Mat(){memset(m,0,sizeof(m));}
};
Mat multi(const Mat &a,const Mat &b)
{
    Mat c;
    for(int i=0;i<x;i++)
    for(int j=0;j<x;j++)if(a.m[j][i]!=0)
    for(int k=0;k<x;k++)if(b.m[i][k]!=0)
    c.m[j][k]=(c.m[j][k]+a.m[j][i]*b.m[i][k]%MOD)%MOD;
    return c;
}
Mat pow(Mat &a,int k)
{
    Mat b;
    for(int i=0;i<x;i++) b.m[i][i]=1;
    while(k)
    {
        if(k&1) b=multi(b,a);
        a=multi(a,a);
        k>>=1;
    }
    return b;
}
int main()
{
    while(~scanf("%d%d%d%d",&n,&b,&K,&x))
    {
        for(int i=0;i<n;++i) scanf("%d",&a[i]);
        Mat A;
        for(int i=0;i<x;++i)
        for(int j=0;j<n;++j)
            ++A.m[i][(i*10+a[j])%x];
        A=pow(A,b);
        printf("%lld\n",A.m[0][K]);
    }
    return 0;
}