【BZOJ1799】【AHOI2009】self 同类分布 题解

题面：传送门

---

    题面这么裸的计数题一看就知道是数位$dp$了。

    由于数字总和不超过$9\times 18=162$，所以我们可以直接枚举。

    我的$dp$状态可能比较繁琐，$dp_{i,j,k,l=0\space or\space 1}$表示当前在第i位，数字总和为$j$，数字大小对数字总和的目标（直接枚举）取模的余数为$k$，前$i$位数字是否与原数相同的状态为$l$（为了确定上界，$l=1$表示相同，反之不同）的总方案数。然后分别算出$1$到$a-1$和$1$到$b$的答案，相减即可。

    时间复杂度：$O(9^4·lg^4a+9^4·lg^4b)$

    还是不理解的看下代码哈：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring> 
using namespace std;
long long dp[20][200][200][2];
long long p[20];
long long calc(long long x)
{
    int s=0;
    long long tmp=x;
    do
    {
        tmp/=10LL;
        s++;
    }
    while(tmp);
    long long res=0;
    for(int d=1;d<=9*18;d++)
    {
        if(s*9<d)break;
        for(int i=1;i<=s;i++)
        {
            for(int k=0;k<=min(i*9,d);k++)
            {
                for(int l=0;l<d;l++)dp[i][k][l][0]=dp[i][k][l][1]=0;
            }
        }
        dp[0][0][0][1]=1;
        for(int i=1;i<=s;i++)
        {
            int cur=x/p[s-i]%10;
            for(int j=0;j<=9;j++)
            {
                for(int k=0;k<=min((i-1)*9,d);k++)
                {
                    for(int l=0;l<d;l++)
                    {
                        if(j<cur)
                        {
                            dp[i][k+j][(l*10+j)%d][0]+=dp[i-1][k][l][0]+dp[i-1][k][l][1];
                        }
                        else if(j==cur)
                        {
                            dp[i][k+j][(l*10+j)%d][0]+=dp[i-1][k][l][0];
                            dp[i][k+j][(l*10+j)%d][1]+=dp[i-1][k][l][1];
                        }
                        else
                        {
                            dp[i][k+j][(l*10+j)%d][0]+=dp[i-1][k][l][0];
                        }
                    }
                }
            }
        }
        res+=dp[s][d][0][0]+dp[s][d][0][1];
    }
    return res;
}
long long a,b;
int main()
{
    p[0]=1;
    for(int i=1;i<=18;i++)p[i]=p[i-1]*10LL;
    scanf("%lld%lld",&a,&b);
    printf("%lld",calc(b)-calc(a-1));
    return 0;
}