40%:直接 DP
刚读题的时候第一感觉:这不是很像我在 《<数谜>解题报告》 里面提到的弱化版问题嘛,搞个无脑 DP,做法很显然。然而我却忽略了最重要的数据范围。
记 f[i][j] 为前 i 层组成的数
f[i+1][(10j+Ak)modY]=f[i+1][(10j+Ak)modY]+f[i][j]
直接根据这个方程进行转移,时间复杂度是 O(NBY) 的。
参考代码:
//2217.cpp
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
using namespace std;
const long long Aspe = 1000000007;
const long long MAXN = 5e4 + 100;
const long long MAXY = 100 + 10;
long long N, B, Z, Y;
long long A[MAXN], f[2][MAXY];
pair <long long, long long> C[MAXY];
int main(void) {
freopen("2217.in", "r", stdin);
freopen("2217.out", "w", stdout);
// scanf("%d%d%d%d", &N, &B, &Z, &Y);
cin >> N >> B >> Z >> Y;
for (long long i = 0; i < N; i++) { cin >> A[i]; /*scanf("%d", &A[i]);*/ ++f[0][A[i] %= Y]; }
sort(A, A + N);
long long cur = 0, cnt = 0, M = 0;
for (long long i = 0; i < N; i++)
if (A[i] == A[cur]) ++cnt; else { C[M++] = make_pair(A[cur], cnt); cur = i; cnt = 1; }
C[M++] = make_pair(A[cur], cnt);
// f[0][0] = 1;
for (long long i = 0; i + 1 < B; i++)
for (long long j = 0; j < Y; j++)
if (f[i & 1][j]) {
for (long long k = 0; k < M; k++) (f[(i + 1) & 1][(j * 10 + C[k].first) % Y] += f[i & 1][j] * C[k].second % Aspe) %= Aspe;
f[i & 1][j] = 0;
}
// printf("%d\n", f[(B - 1) & 1][Z]);
cout << f[(B - 1) & 1][Z] << endl;
return 0;
}
100%:
上面的 DP 并没有大的问题,但就是效率低下,需要考虑对其进行优化。常见的 DP 优化手段?降维?不能再压了。单调性和斜率优化,这题显然不满足。
此时就需要用到矩阵乘法。接下来以样例 1 进行说明。
不妨将 f[i] 的值保存在 1 行 Y 列的矩阵 A 中:
A=[f[i][0]f[i][1]f[i][2]f[i][3]f[i][4]f[i][5]f[i][6]f[i][7]f[i][8]