CF: 341-Div2 E ( Matrix Expo )

CF: 341-Div2 E ( Matrix Expo )

///==================================================///
/// HELLO WORLD !! ///
/// IT'S ME ///
/// BISHAL GAUTAM ///
/// [ bsal.gautam16@gmail.com ] ///
///==================================================///
#include<bits/stdc++.h>
#define X first
#define Y second
#define mpp make_pair
#define nl printf("\n")
#define SZ(x) (int)(x.size())
#define pb(x) push_back(x)
#define pii pair<int,int>
#define pll pair<ll,ll>
///---------------------
#define S(a) scanf("%d",&a)
#define P(a) printf("%d",a)
#define SL(a) scanf("%lld",&a)
#define S2(a,b) scanf("%d%d",&a,&b)
#define SL2(a,b) scanf("%lld%lld",&a,&b)
///------------------------------------
#define all(v) v.begin(),v.end()
#define CLR(a) memset(a,0,sizeof(a))
#define SET(a) memset(a,-1,sizeof(a))
#define fr(i,a,n) for(int i=a;i<=n;i++)
using namespace std;
typedef long long ll;

/// Digit 0123456789012345678 ///
#define MX 500002
#define inf 2000000010
#define MD 1000000007
#define eps 1e-9
///===============================///

#define N 102
int ms;///Matrix Size GLOBAL: N
struct mat {
ll m[N][N];
mat() {};
void Nullify( ) {
for(int i=0; i<ms; i++) {
for(int j=0; j<ms; j++) {
m[i][j]=0;
}
}
}
void Unify( ) {
for(int i=0; i<ms; i++) {
for(int j=0; j<ms; j++) {
m[i][j]=(i==j);
}
}
}
mat operator *(const mat &b )const {
mat c;
for(int i=0; i<ms; i++) {
for(int j=0; j<ms; j++) {
c.m[i][j]=0;
for(int k=0; k<ms; k++) {
c.m[i][j]=( c.m[i][j]+ (m[i][k]*b.m[k][j])%MD )%MD;///FOR THIS PROB
}
}
}
return c;
}
mat Pow(int p) {
mat ret;
ret.Unify();
mat B=(*this);
for(; p; p>>=1) {
if(p & 1)ret = (ret * B);
B = (B * B);
}
return ret;
}
void Print( ) {
cout<<endl;
cout<< "Matix is:\n";
cout<< "****************************\n";
for(int i=0; i<ms; i++) {
for(int j=0; j<ms; j++) {
cout<<m[i][j]<< " ";
}
cout<<endl;
}
cout<< "****************************\n";
cout<<endl;
}
} b;

int Fr[12];
int main() {
int i,j,n,m,k,x,y;
S2(n,m);
S2(k,x);
for(i=1; i<=n; i++) {
S(y);
Fr[ y ]++;
}
ms=100;
b.Nullify();
for(i=0; i<100; i++) {
for(j=1; j<=9; j++) {
int nx=((i*10)+j)%x;
b.m[ i ][ nx ]+=Fr[j];
}
}
mat ret=b.Pow(m);
ll ans=ret.m[ 0 ][ k ];
printf("%lld\n",ans);
return 0;
}

Comments

Popular posts from this blog

HackerEarth: City and Campers 2 (DSU-UnionFind)

Two Pointers Technique & Binary Search For Beginners

Problem : Codeforces Round #406 (Div. 1) B [ Legacy ]( Dijakstra, Segment tree)