Codeforces 621E Wet Shark and Block【dp + 矩阵快速幂】

题意：

有b个blocks，每个blocks都有n个相同的0~9的数字，如果从第一个block选1，从第二个block选2，那么就构成12，问对于给定的n,b有多少种构成方案使最后模x的余数为k。

分析：

dp+矩阵快速幂。
假如现在的数是m，模x余数是n，那么再从下一个block中选一个数a，a模x余数为b，那么新的数的余数就为(m∗10+a)%x，也就是(n∗10+b)%x，所以实际上我们只需要直接对余数进行操作。容易得到状态转移方程，其中dp[i][j]表示从第i个block中选择一个数后，余数为j的方案数，cnt[m]为余数为m的数的个数。

dp[i][(j * 10 + m) % x] = dp[i-1][j] * cnt[m];

可是b高达109，规模太大直接递推的话效率太低，而x最大仅为100，直接用矩阵表示这个递推式，时间复杂度则降为O(x3logb)。

代码：

#include<cstdio>
const int maxn = 50005;
int cnt[maxn], r[maxn];
const int N = 105, mod = 1e9 + 7;
struct Matrix
{
    int row,cal;
    long long  m[N][N];
};
Matrix init(Matrix a, long long t)
{
    for(int i = 0; i < a.row; i++)
        for(int j = 0; j < a.cal; j++)
            a.m[i][j] = t;
    return a;
}
Matrix mul(Matrix a,Matrix b)
{
    Matrix ans;
    ans.row = a.row, ans.cal = b.cal;
    ans = init(ans,0);
    for(int i = 0; i < a.row; i++)
        for(int j = 0; j < b.cal; j++)
            for(int k = 0; k < a.cal; k++)
                ans.m[i][j] = (ans.m[i][j] + a.m[i][k] * b.m[k][j])%mod;
    return ans;
}
long long quick_pow(int k, int x, int res, Matrix A)
{
    Matrix I;
    I.row = x, I.cal = 1;
    I = init(I, 0);
    for(int i = 0; i < x; i++)
            I.m[i][0] = cnt[i];
    while(k){
        if(k&1) I = mul(A, I);
        A = mul(A, A);
        k>>=1;
    }
    return I.m[res][0]%mod;
}
int main (void)
{
    int n, b, k, x;
    int a;
    scanf("%d%d%d%d",&n,&b,&k,&x);
    for(int i = 0; i < n; i++){
        scanf("%d",&a);
        cnt[a%x]++;
    }
    for(int i = 0; i < x; i++)
        r[i] = (i *10)%x;

    Matrix t;
    t.row = t.cal = x;
    for(int i = 0; i < t.row; i++)
        for(int j = 0; j < t.cal; j++)
            t.m[i][j] = cnt[(i+x-r[j])%x];

    printf("%I64d",quick_pow(b-1, x, k, t));
    return 0;

}

转载于:https://www.cnblogs.com/Tuesdayzz/p/5758778.html