AtCoder agc009_e题解

题目大意：

给定$n$个0和$m$个1，每次可以选择$k$个数删去它们，然后把它们的平均数写上去。

求最后剩下的数有几种。

答案对$1e9+7$取模

保证$n+m-1$能被$k$整除

$sol:$

把这个问题转化为$k$叉树的模型。

其有$n+m$个叶子结点。

其中有$n$个写着0，$m$个写着1。

对于非叶子节点，它的值就是它的儿子节点的平均值。

假设这$n$个0的深度分别是$d{n}_i$，$m$个1的深度为$d{m}_i$，那么根节点的数就是
$$\sum _{i=1}^m{\left(\frac{1}{k}\right)}^{d{m}_i}$$
显然，如果所有叶节点都是1的话那么根节点也是1，所以
$$\sum _{i=1}^m{\left(\frac{1}{k}\right)}^{d{m}_i}+\sum _{i=1}^n{\left(\frac{1}{k}\right)}^{d{n}_i}=1$$
若满足这个条件的话，一定能构造出合法的$k$叉树

现在问题成了求有多少个$q$能写成$n$个$(\frac{1}{k}{)}^{dn}$，多少个$1-q$能写成$m$个$(\frac{1}{k}{)}^{dm}$相加的形式。

将根节点的权值用$k$进制小数表示如同$0,z_1,z_2,z_3,\cdots ,z_n,0\le z_i<k$

不考虑进位的话$\sum z=m$

考虑上进位的话，每次可以将$z_i$减去$1$，让$z_{i+1}$加上$k$

则$\sum z\equiv m(\mathrm{m}\mathrm{o}\mathrm{d}k-1)$

假设小数有$len$位，则$1-q$的位数和应该是$(len-1)(k-1)+k-\sum z=len(k-1)-\sum z+1$

设$f[i][j]$表示小数点后第$i$位，每一位的和是$j$

直接dp就行了。

由于小数的末尾不能是0，所以当第$i$位的末尾是否是0也应该加进状态里面。

#include<bits/stdc++.h>
using namespace std;
#define f1(a,b,c) for(int c=a;c<=b;c++)
#define f2(a,b,c) for(int c=a;c>=b;c--)
#define f3(a,b,c) for(int c=a;c;c=b)
#define so1(a,n) sort(a+1,a+n+1,mycmp);
#define so2(a,n) sort(a+1,a+n+1);
#define ll long long
#define itn int
#define ubt int 
#define mp make_pair
#define pii pair<int,int>
#define pll pair<ll,ll>
const int twx=2e3+100;
const int MOD=1e9+7;
const int inf=0x3f3f3f3f;
ll read()
{
    ll sum=0;
    ll flag=1;
    char c=getchar();
    while(c<'0'||c>'9')
    {
        if(c=='-')
        {
            flag=-1;
        }
        c=getchar();
    }
    while(c>='0'&&c<='9')
    {
        sum=((sum*10)+c-'0');
        c=getchar();
    }
    return sum*flag;
}
int n,m,k;
int asd[twx<<1][twx][2];
int ans;
int sum[twx];
ll add(ll x,ll y)
{
    return x+y>=MOD?x+y-MOD:x+y;
}
ll sub(ll x,ll y)
{
    return x-y<0?x-y+MOD:x-y;
}
ll mul(ll x,ll y)
{
    return 1LL*x*y%MOD;
}
void init()
{
	n=read();
    m=read();
    k=read();
}
void work()
{
    asd[0][0][0]=1;
    f1(1,n+m,i)
    {
        sum[0]=add(asd[i-1][0][0],asd[i-1][0][1]);
        f1(1,n,j)
        {
            sum[j]=add(sum[j-1],add(asd[i-1][j][0],asd[i-1][j][1]));
        }
        f1(0,n,j)
        {
            asd[i][j][0]=add(asd[i-1][j][0],asd[i-1][j][1]);
            if(j)
            {
                asd[i][j][1]=sub(sum[j-1],(j-k>=0?sum[j-k]:0));
            }
        }
        f1(0,n,j)
        {
            if(j%(k-1)==n%(k-1)&&(i*(k-1)-j+1)%(k-1)==m%(k-1)&&i*(k-1)-j+1<=m)
            {
                ans=add(ans,asd[i][j][1]);
            }
        }
    }
}
void print()
{
	printf("%d\n",ans);
}
int main()
{
    init();
    work();
    print();
    return 0;
}