LuoguP4127(数位dp)

链接:点击打开链接

题意:给出两个数 a,b,求出[a,b]中各位数字之和能整除原数的数的个数。
代码:

#include 
using namespace std;
typedef long long ll;

ll t,a[20],dp[20][180][180];
ll dfs(ll pos,ll sum,ll mod,ll lim,ll tmp){
    ll i,u,ans=0;
    if(pos==0){
        if(sum==t&&mod==0)
        return 1;
        return 0;
    }
    if(lim==0&&dp[pos][sum][mod]!=-1)
    return dp[pos][sum][mod];
    u=lim?a[pos]:9;
    for(i=0;i<=u;i++){
        if(sum+i>t)
        continue;
        ans+=dfs(pos-1,sum+i,(mod*10+i)%t,lim&(i==u),tmp*10+i);
    }
    if(lim==0)
    dp[pos][sum][mod]=ans;
    return ans;
}
ll cal(ll x){                                   
    ll pos=0;
    while(x){
        a[++pos]=x%10;
        x/=10;
    }
    return dfs(pos,0,0,1,0);
}
int main(){
    ll l,r,ans=0;
    scanf("%lld%lld",&l,&r);
    for(t=1;t<=9*18;t++){                       //枚举每一位的和的可能,然后数位dp
        memset(dp,-1,sizeof(dp));
        ans+=(cal(r)-cal(l-1));
    }
    printf("%lld\n",ans);
    return 0;
}

 

你可能感兴趣的:(---------数位dp,洛谷,数位dp)