AGC009：Eternal Average

传送门
好神啊
直接考虑一棵 \(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;
}

标签：int,AGC009,Eternal,deep,sum,inline,equiv,Average,mod
来源： https://www.cnblogs.com/cjoieryl/p/10357644.html