首页 > 其他 > 详细

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

时间:2019-08-04 14:32:38      阅读:65      评论:0      收藏:0      [点我收藏+]
#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 烦人的数学作业

#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(){
    int t;input(t);
    while(t--){
        memset(ans,0,sizeof(ans));
        memset(dp,0,sizeof(dp));
        memset(add,0,sizeof(add));
        now=0;
        int out=0;
        input(a);input(b);
        solve(a-1);solve(b);
        for(int i=0;i<=9;i++){
            out+=i*((ans[2][i]-ans[1][i])%ll(1e9+7));
            out%=ll(1e9+7);
        }
        writeln(out);
    }
}

注意:

        for(int i=0;i<=9;i++){
            out+=i*((ans[2][i]-ans[1][i])%ll(1e9+7));
            out%=ll(1e9+7);
        }

有两处模,注意到了吗!
这个要注意!

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

原文:https://www.cnblogs.com/Y15BeTa/p/11297675.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!