LuoguP2602 [ZJOI2010]数字计数(数位dp)

#include<cstdio>
#include<cstring>
#include<iostream>
#define int long long
using namespace std;
typedef long long ll;

int a,b,ans[3][10],dp[20][2],add[20][2][10],shu[20],now;

inline void input(ll &x){
    ll ans=0,f=1;
    char c=getchar();
    while(c>'9'||c<'0'){
        if(c=='-')f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9'){
        ans=ans*10+c-48;
        c=getchar();
    }
    x=ans*f;
}

inline void output(ll x){
    if(x<0)x=-x,putchar('-');
    if(x>9)output(x/10);
    putchar(x%10+48);
}

inline void writeln(ll x){
    output(x);
    putchar('\n');
}

inline int dfs(int len,int limit,int have){
    if(len==0)return 1;
    if(!limit&&dp[len][have]){
    //cout<<len<<' '<<limit<<' '<<have<<' '<<dp[len][have]<<endl;
        for(int i=0;i<=9;i++){
            ans[now][i]+=add[len][have][i];
        }
        return dp[len][have];
    }
    int maxi=limit?shu[len]:9,cnt=0,yuan[10];
    for(int i=0;i<=9;i++)yuan[i]=ans[now][i];
    for(int i=0;i<=maxi;i++){
        int x=dfs(len-1,limit&&i==maxi,i!=0||have);
        if(i!=0||have)ans[now][i]+=x;
        cnt+=x;
    }
    if(!limit){
        for(int i=0;i<=9;i++)add[len][have][i]=ans[now][i]-yuan[i];
        dp[len][have]=cnt;
    }
    return cnt;
}

inline void solve(int x){
    now++;
    memset(shu,0,sizeof(shu));
    int tot=0;
    while(x){
        shu[++tot]=x%10;
        x/=10;
    }
    dfs(tot,1,0);
}

signed main(){
    input(a);input(b);
    solve(a-1);solve(b);
    for(int i=0;i<=9;i++){
        //printf("%I64d %I64d %I64d\n",ans[1][i],ans[2][i],ans[2][i]-ans[1][i]);
        output(ans[2][i]-ans[1][i]);putchar(' ');
    }
}

记得记忆化搜索。
(把<=9打成<9了调了半天QwQ)
双倍经验:P4999 烦人的数学作业

猜你喜欢

转载自www.cnblogs.com/Y15BeTa/p/11297675.html