[agc009e]Eternal Average

前言

需要发现性质的dp。
重点在于发现性质。

题意

请构造若干k叉树，使得每颗均有n+m个叶子，有n个叶子点权为0，m个叶子点权为1，对于非叶子点权为儿子点权的平均数。
问最多构造多少k叉树，使得任意两颗根节点点权不同？

DP

令n个叶子的深度为x1~xn，同理m个叶子的深度为y1~ym。
发现根节点的点权是$\sum_{i=1}^mk^{-y_i}$。
如果所有叶子点权都为1，那么根节点点权一定为1，因此有
$\sum_{i=1}^nk^{-x_i}+\sum_{i=1}^mk^{-y_i}=1$
设答案为z，那么z能被表示成m个k的次幂的数的和。
同时1-z能被表示成n个k的次幂的数的和。
我们探索一个数z什么时候能够被表示成n个k的次幂的数的和。
每一位的数字和一定不能超过n，而且要与n在模k-1下同余。
这个结论易证。
z和1-z的最低位一定相同，因此可以把m减一，那么总和在k进制下每一位都是k-1。
然后现在问题变成构造一个序列$z_{1-l}$，满足下列条件。
对于$1\leq i\leq l$有$0\leq z_i\leq k-1$。
$z_l>0$
$\sum_{i=1}^lz_i\leq n$且$\sum_{i=1}^lz_i\equiv n(mod\ k-1)$
$\sum_{i=1}^l(k-1-z_i)\leq m-1$且$\sum_{i=1}^l(k-1-z_i)\equiv m-1(mod\ k-1)$
可以简单设个dp然后用前缀和优化解决这个问题。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=2000+10,mo=1000000007;
int f[2][maxn][2],sum[maxn];
int i,j,k,l,t,n,m,ans,now;
int main(){
    scanf("%d%d%d",&n,&m,&k);
    m--;k--;
    fo(i,1,k) f[0][i][1]=1;
    f[0][0][0]=1;
    fo(i,0,n)
        if (k-i<=m&&i%k==n%k&&(k-i)%k==m%k) (ans+=f[0][i][1])%=mo;
    now=0;
    fo(i,2,max(n,m)*2){
        now=1-now;
        fo(j,0,n) sum[j+1]=(sum[j]+(f[1-now][j][0]+f[1-now][j][1])%mo)%mo;
        fo(j,0,n){
            f[now][j][0]=(sum[j+1]-sum[j])%mo;
            l=max(j-k,0);
            f[now][j][1]=(sum[j]-sum[l])%mo;
        }
        fo(j,0,n)
            if (k*i-j<=m&&j%k==n%k&&(k*i-j)%k==m%k) (ans+=f[now][j][1])%=mo;
    }
    (ans+=mo)%=mo;
    printf("%d\n",ans);
}