self 同类分布 【HYSBZ - 1799】【数位dp】

题目链接

给出a,b，求出[a,b]中各位数字之和能整除原数的数的个数。

刚开始的时候题目看错了，以为是【a，b】所有数的和，能整除这个区间的数有几个。然后数据范围还这么大，就开始懵逼，后来发现题看错了。
其实这个题是个数位dp，因为各位数的和最多有个9*17所以，我们去枚举一下各位数的和，然后去dp。看有多少个数可以整除当前数位的和。
其他的就是正常的数位dp啦。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
long long dp[25][200][200];
int a[50];
long long dfs(int pos,int p,int sta1,int sta2,int limit)
{
    if(pos==-1)
        return (sta1==0&&sta2==p);
    if(sta2>p||sta2+(pos+1)*9<p)
        return 0;
    if(!limit&&dp[pos][sta1][sta2]!=-1)
        return dp[pos][sta1][sta2];
    long long res=0;
    int up=limit?a[pos]:9;
    for(int i=0;i<=up;i++)
    {
        res+=dfs(pos-1,p,(sta1*10+i)%p,sta2+i,limit&&i==a[pos]);
    }
    if(limit)
        return res;
    return dp[pos][sta1][sta2]=res;
}
long long solve(long long x)
{
    int pos=0;
    while(x)
    {
        a[pos++]=x%10;
        x/=10;
    }
    long long res=0;
    for(int i=1;i<=pos*9;i++)
    {
        memset(dp,-1,sizeof(dp));
        res+=dfs(pos-1,i,0,0,1);
    }
    return res;
}
int main()
{
    long long a,b;
    while(~scanf("%lld %lld",&a,&b))
    {
        printf("%lld\n",solve(b)-solve(a-1));
    }
    return 0;
}