水题数位dp 0~9

来源:互联网 发布:影楼后期排版软件 编辑:程序博客网 时间:2024/06/05 17:50

题意:记录a到b 的数字,0~9的出现的次数
数位dp水题,但是自己找了很久没找到错,
最后才发现是因为没有加sta位置的状态错了。。。

LL dp[20][20][15][20];int num[30];LL dfs(int sta,int pos,int limit,int dig,int cnt){    if(pos<0){return cnt;}    if(!limit&&~dp[sta][pos][dig][cnt])return dp[sta][pos][dig][cnt];    int up=limit?num[pos]:9;    LL ret=0;    for(int i=0;i<=up;++i){        if(sta==pos&&i==0&&dig==0)ret+=dfs(sta-1,pos-1,limit&&i==up,dig,cnt);        //else if(sta!=pos&&i==0&&dig==0)ret+=dfs(sta,pos-1,limit&&i==up,0,cnt+1);        else ret+=dfs(sta,pos-1,limit&&i==up,dig,cnt+(i==dig?1:0));        if(dig==0){            for(int j=sta;j<=pos;++j)cout<<num[j];cout<<i;puts("");        }    }    if(!limit)dp[sta][pos][dig][cnt]=ret;    return ret;}LL ans[20];LL solve(LL x,int dig){    int pos=0;    while(x){        num[pos++]=x%10;        x/=10;    }    return dfs(pos-1,pos-1,1,dig,0);}int main(){    LL a,b;    mem(dp,-1);    while(~sf("%lld%lld",&a,&b)){        for(int i=0;i<=9;++i)ans[i]=solve(b,i)-solve(a-1,i);        pf("%lld %lld\n",solve(b,0),solve(a,0));        for(int i=0;i<=9;++i)pf("%lld%c",ans[i],i==9?'\n':' ');    }}
原创粉丝点击