Codeforces 621E Wet Shark and Blocks【Dp+矩阵快速幂】

来源:互联网 发布:淘粉吧和淘宝的关系 编辑:程序博客网 时间:2024/05/22 08:15

E. Wet Shark and Blocks
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

There are b blocks of digits. Each one consisting of the samen digits, which are given to you in the input. Wet Shark must chooseexactly one digit from each block and concatenate all of those digits together to form one large integer. For example, if he chooses digit1 from the first block and digit 2 from the second block, he gets the integer 12.

Wet Shark then takes this number modulo x. Please, tell him how many ways he can choose one digit from each block so that he gets exactlyk as the final result. As this number may be too large, print it modulo109 + 7.

Note, that the number of ways to choose some digit in the block is equal to the number of it's occurrences. For example, there are3 ways to choose digit 5 from block3 5 6 7 8 9 5 1 1 1 1 5.

Input

The first line of the input contains four space-separated integers, n, b, k and x (2 ≤ n ≤ 50 000, 1 ≤ b ≤ 109, 0 ≤ k < x ≤ 100, x ≥ 2) — the number of digits in one block, the number of blocks, interesting remainder modulo x and modulox itself.

The next line contains n space separated integersai (1 ≤ ai ≤ 9), that give the digits contained in each block.

Output

Print the number of ways to pick exactly one digit from each blocks, such that the resulting integer equalsk modulo x.

Examples
Input
12 1 5 103 5 6 7 8 9 5 1 1 1 1 5
Output
3
Input
3 2 1 26 2 2
Output
0
Input
3 2 1 23 1 2
Output
6
Note

In the second sample possible integers are 22, 26, 62 and 66. None of them gives the remainder1 modulo 2.

In the third sample integers 11, 13, 21, 23, 31 and 33 have remainder 1 modulo 2. There is exactly one way to obtain each of these integers, so the total answer is6.


题目大意:

给你N个数,b个块,每个块都是n个数,且都一样,现在让你从每个块取一个数,并且按序拼接在一起之后%x==k的方案数。

按序拼接:第一个块取的数是i,第二个块取得数是j,第三个块取的数是k,那么按序拼接在一起得到的数就是ijk.


思路:


1、

统计方案数,考虑dp,不难想到设定dp【i】【j】表示进行到第i个块,所取数据%x==j的方案数。

那么就有:

dp【i】【(j*10+l)%m】+=dp【i-1】【j】*num【l】.表示当前这个块取了数字l拼接到最后的方案数转移情况。

观察到b有1e9辣么大。那么可以使用矩阵快速幂进行优化。


2、矩阵的设定:

对于第一个矩阵中的数据,要根据实际情况决定。



Ac代码:


#include<stdio.h>#include<string.h>using namespace std;#define mod 1000000007#define ll __int64typedef struct Matrix{    ll mat[102][102];}matrix;matrix A,B,tmp;ll num[15];ll n,b,x,m;Matrix matrix_mul(matrix a,matrix b){    matrix c;    memset(c.mat,0,sizeof(c.mat));    ll i,j,k;    for(ll i=0;i<m;i++)    {        for(ll j=0;j<m;j++)        {            for(ll k=0;k<m;k++)            {                c.mat[i][j]=(c.mat[i][j]+(a.mat[i][k]*b.mat[k][j])%mod)%mod;            }        }    }    return c;}Matrix matrix_quick_power(matrix a,ll k)//矩阵快速幂0.0{    matrix b;    memset(b.mat,0,sizeof(b.mat));    for(ll i=0;i<m;i++)    b.mat[i][i]=1;//单位矩阵b    while(k)    {        if(k%2==1)        {            b=matrix_mul(a,b);            k-=1;        }        else        {            a=matrix_mul(a,a);            k/=2;        }    }    return b;}int main(){    while(~scanf("%I64d%I64d%I64d%I64d",&n,&b,&x,&m))    {        memset(tmp.mat,0,sizeof(tmp.mat));        memset(A.mat,0,sizeof(A.mat));        memset(num,0,sizeof(num));        for(ll i=1;i<=n;i++)        {            ll x;scanf("%I64d",&x);            num[x]++;        }        for(ll i=0;i<m;i++)        {            for(ll j=0;j<m;j++)            {                for(ll l=1;l<=9;l++)                {                    ll tmpnum=j*10+l;                    tmpnum%=m;                    if(tmpnum==i)                    {                        A.mat[i][j]+=num[l];                    }                }            }        }        for(ll i=0;i<m;i++)        {            for(ll j=1;j<=9;j++)            {                if(j%m==i)                {                    tmp.mat[i][0]+=num[j];                }            }        }        B=matrix_quick_power(A,b-1);        B=matrix_mul(B,tmp);        printf("%I64d\n",B.mat[x][0]);    }}








0 0