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; }
上一篇:hbase搭建


下一篇:HDU 2081 手机短号