CF 1073 E. Segment Sum

https://codeforces.com/problemset/problem/1073/E

题意：[l,r]中，出现0—9数字的种类数不超过k的数的和

dp[i][j][0/1] 表示 dfs到第i位以后，数字出现的情况状态为j能转移的状态，是否有上界限制的数字的和

f[i][j] 表示 dfs到第i位以后，数字出现的情况状态为j能转移的状态，没有上界限制的数字的个数

枚举这一位填什么

加上这一位的贡献，加上填上这一位之后后面数的贡献

数字出现状态：若i出现过，则状态j的第i位二进制位为1

#include<cstdio>
#include<cstring>

using namespace std;

typedef long long LL;

const int mod=;

int dp[][(<<)+][],f[][(<<)+];
int ans,m;
int a[],pow10[];
bool va[(<<)+];

int k;

LL dfs(int dep,int s,int lim,bool zero)
{
    if(!va[s]) return ;
    if(!lim && f[dep][s]!=-) return f[dep][s];
    if(!dep) return ;
    int up=lim ? a[dep] : ,nl,news,tmp;
    int sum=;
    for(int i=;i<=up;++i)
    {
        if(zero && !i) news=;
        else news=s|<<i;
        nl=lim && a[dep]==i;
        tmp=dfs(dep-,news,nl,zero && !i);
        dp[dep][s][lim]+=1ll*tmp*i%mod*pow10[dep-]%mod;
        dp[dep][s][lim]%=mod;
        dp[dep][s][lim]+=dp[dep-][news][nl];
        dp[dep][s][lim]%=mod;
        sum+=tmp;
        sum%=mod;
    }
    if(!lim) f[dep][s]=sum;
    return sum;
}

void solve(long long n,int ty)
{
    int len=;
    while(n) a[++len]=n%,n/=;
    memset(f,-,sizeof(f));
    memset(dp,,sizeof(dp));
    dfs(len,,,);
    ans+=ty*dp[len][][];
    ans%=mod;
    if(ans<) ans+=mod;
}

bool check(int s)
{
    int p=;
    for(int i=;i<=;++i)
        if(s&<<i) p++;
    if(p<=k) return ;
    else return ;
}

int main()
{
    LL l,r;
    scanf("%lld%lld%d",&l,&r,&k);
    pow10[]=;
    for(int i=;i<;++i) pow10[i]=1ll*pow10[i-]*%mod;
    m=<<;
    for(int i=;i<m;++i) va[i]=check(i);
    solve(r,);
    solve(l-,-);
    printf("%d",ans);
}