CodeForces 621E Wet Shark and Blocks

CodeForces 621E Wet Shark and Blocks

矩阵快速幂

传送门：CodeForces

传送门：HustOJ

---

题意

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

---

思路

如果不考虑b的大小，我们能想到dp，dp[k][(i*10+j)%x]+=dp[k-1][i]*num[j] (1<=j<=9)，dp[k][i]表示现在正在考虑前k个块，此时结果取模x等于i的方案数。
但是我们发现，b实在是太大了，但是线性递推都是能用快速矩阵幂来搞的，这个也不例外，我们直接根据方程构造出矩阵，然后用快速矩阵幂就能得到答案了。
因为得到的数是模x，所以一共有0到x-1的x种可能性。
用一个矩阵x*1的矩阵存结果，表示这x种答案的方案数，矩阵第一个数为1，其余为0
然后构造系数矩阵，显然系数矩阵第i行第j列代表着通过一个block把答案从j变成i的方案数。所以只要先求出系数矩阵再用矩阵快速幂即可。

---

代码

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define _ ios_base::sync_with_stdio(0);cin.tie(0);
#define M(a,b) memset(a,b,sizeof(a))
using namespace std;

const int MAXN=105;
const int oo=0x3f3f3f3f;
typedef __int64 LL;
const LL loo=4223372036854775807ll;
typedef long double LB;
const LL mod=1e9+7;

int num[MAXN];
struct Matrix
{
    static const int N=105;
    LL ma[N][N];
    Matrix() { M(ma, 0); }
    Matrix(const Matrix &m)
    {
        for(int i=0;i<N;i++)
            for(int j=0;j<N;j++)
                ma[i][j]=m.ma[i][j];

    }
    Matrix operator * (const Matrix &m)
    {
        Matrix res;
        for(int i=0;i<N;i++)
        {
            for(int j=0;j<N;j++)
            {
                for(int k=0;k<N;k++)
                {
                    res.ma[i][j]=(res.ma[i][j]+ma[i][k]*m.ma[k][j])%mod;
                }
            }
        }
        return res;
    }
    Matrix operator ^ (int n)//注意顺序
    {
        Matrix res;
        Matrix tmp=(*this);
        for(int i=0;i<N;i++) res.ma[i][0]=num[i];
        while(n)
        {
            if(n%2)
            {
                res=tmp*res;
            }
            n=n/2;
            tmp=tmp*tmp;
        }
        return res;
    }
};
int main()
{
    _
    int n;
    while(cin>>n)
    {
        int b, k, x;cin>>b>>k>>x;
        M(num, 0);
        for(int i=0;i<n;i++)
        {
            int temp;cin>>temp;
            num[temp%x]++;
        }
        Matrix ma;
        for(int i=0;i<x;i++)
        {
            for(int j=0;j<x;j++)
            {
                ma.ma[i][j]=num[(i+x-((10*j)%x))%x];
            }
        }
        Matrix res=(ma^(b-1));
        cout<<res.ma[k][0]%mod<<endl;
    }
    return 0;
}