[luogu4127 AHOI2009] 同类分布 (数位dp)

Posted 2021-01-13 menteur-hxy

传送门

Solution

裸数位dp，空间存不下只能枚举数字具体是什么
注意memset最好为-1，不要是0，有很多状态答案为0

Code

//By Menteur_Hxy
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define Re register
#define Fo(i,a,b) for(Re int i=(a),_=(b);i<=_;i++)
#define Ro(i,a,b) for(Re int i=(b),_=(a);i>=_;i--)
#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin)),p1==p2?EOF:*p1++)
using namespace std;
typedef long long LL;

char buf[1<<21],*p1,*p2;
inline LL read() {
    LL x=0,f=1;char c=getchar();
    while(!isdigit(c)) {if(c=='-')f=-f;c=getchar();}
    while(isdigit(c)) x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x*f;
}

LL MOD;
LL dp[20][200][200],bit[20];

LL dfs(LL pos,LL sum,LL mod,LL lim) {
    if(!pos) return (sum==MOD&&mod==0);
    if(!lim&&~dp[pos][sum][mod]) return dp[pos][sum][mod];
    int up=lim?bit[pos]:9; LL res=0;
    Fo(i,0,up) res+=dfs(pos-1,sum+i,(mod*10+i)%MOD,lim&&i==bit[pos]);
    if(!lim) dp[pos][sum][mod]=res;
    return res;
}

LL sol(LL x) {
    LL len=0,res=0;
    while(x) bit[++len]=x%10,x/=10;
    for(MOD=1;MOD<=len*9;MOD++) {
        memset(dp,-1,sizeof(dp)); 
        res+=dfs(len,0,0,1);
    }
    return res;
}

int main() {
    LL l=read(),r=read();
    printf("%lld",sol(r)-sol(l-1));
    return 0;
}