Codeforces621 E. Wet Shark and Blocks（矩阵快速幂优化dp）

题意：

解法：

$$\begin{gathered} 令d[i][j]表示前i个格子,当前模x=j的方案数. \\ 如果第i个格子存在数p,那么d[i+1][(j\ast 10+p)\%x]+=d[i][j]. \\ 直接dp复杂度是O(b\ast 100)的. \\ 但是由于每次的转移系数相同,因此可以用矩阵优化到O(log(b)\ast 100^3) \end{gathered}$$

code：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=105;
const int mod=1e9+7;
struct Mat{
    int a[maxm][maxm];
    void init(){memset(a,0,sizeof a);}
    void init_e(){init();for(int i=0;i<maxm;i++)a[i][i]=1;}
    Mat operator*(const Mat& b)const{
        Mat ans;ans.init();
        for(int k=0;k<maxm;k++)
            for(int i=0;i<maxm;i++)if(a[i][k])
                for(int j=0;j<maxm;j++)if(b.a[k][j])
                    ans.a[i][j]=(ans.a[i][j]+a[i][k]*b.a[k][j]%mod)%mod;
        return ans;
    }
};
int cnt[maxm];
int n,b,k,x;
Mat ppow(Mat a,int b){
    Mat ans;ans.init_e();
    while(b){if(b&1)ans=ans*a;a=a*a;b>>=1;}
    return ans;
}
void solve(){
    cin>>n>>b>>k>>x;
    for(int i=1;i<=n;i++){
        int p;cin>>p;cnt[p%x]++;
    }
    Mat base;base.init();
    for(int i=0;i<100;i++){
        for(int j=0;j<100;j++){
            int nt=(i*10+j)%x;
            base.a[nt][i]+=cnt[j];
        }
    }
    Mat pre;pre.init();
    pre.a[0][0]=1;
    pre=ppow(base,b)*pre;
    int ans=pre.a[k][0];
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    solve();
    return 0;
}

---