传送门
好神啊
直接考虑一棵 \(n+m\) 个叶子的 \(k\) 叉树,根结点权值为 \(\sum_{i\in m}(\frac{1}{k})^{deep_i}\)
对于一个 \(deep\) 的序列
如果 \(\sum_{i\in m}(\frac{1}{k})^{deep_i}+\sum_{i\in n}(\frac{1}{k})^{deep_i}=1\)
那么一定可以构造出一棵 \(k\) 叉树满足要求
(从deep大到小考虑,除去 \(k\) 进位)
那么对于一个答案数 \(x\),它的每一位的数字(可以通过退位)和 \(=m\)
退位就是 \(+k-1\),那么也就是 \(\sum\equiv m(mod~k-1)\)
同理 \(1-x\) 的每一位的数字(可以通过退位)和 \(=n\)
设位数为 \(l\)
即 \(l(k-1)+1-\sum\equiv n(mod~k-1)\)
那么可以设 \(f_{i,j}\) 表示小数点后前 \(i\) 位,每一位的数字和为 \(j\),的不同的数字个数
每次判断 \(j\le m,i(k-1)+1-j\le n\) 且 \(j\equiv m(mod~k-1),l(k-1)+1-j\equiv n(mod~k-1)\)
满足则贡献答案
\(i\) 只需要枚举到 \(n+m\) 即可
\(dp\) 只需要前缀和优化,注意要记录后缀 \(0\)
# include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn(2005);
const int mod(1e9 + 7);
inline void Inc(int &x, int y) {
x = x + y >= mod ? x + y - mod : x + y;
}
inline void Dec(int &x, int y) {
x = x - y < 0 ? x - y + mod : x - y;
}
inline int Add(int x, int y) {
return x + y >= mod ? x + y - mod : x + y;
}
inline int Sub(int x, int y) {
return x - y < 0 ? x - y + mod : x - y;
}
int n, m, k, f[maxn << 1][maxn][2], ans, s[maxn][2];
inline int Check(int len, int sum) {
return sum % k == m % k && (len * k + 1 - sum) % k == n % k && len * k + 1 - sum <= n;
}
int main() {
int i, j, r, t;
scanf("%d%d%d", &n, &m, &k), r = n + m, f[0][0][0] = 1, --k;
for (i = 1; i <= r; ++i) {
for (t = 0; t < 2; ++t)
for (s[0][t] = f[i - 1][0][t], j = 1; j <= m; ++j)
s[j][t] = Add(s[j - 1][t], f[i - 1][j][t]);
for (j = 0; j <= m; ++j) f[i][j][1] = Add(f[i - 1][j][0], f[i - 1][j][1]);
for (j = 1; j <= m; ++j)
for (t = 0; t < 2; ++t)
Inc(f[i][j][0], Sub(s[j - 1][t], j > k ? s[j - k - 1][t] : 0));
for (j = 0; j <= m; ++j) if (Check(i, j)) Inc(ans, f[i][j][0]);
}
printf("%d\n", ans);
return 0;
}