hihocoder1033-数位dp&数位求和&好题-交错和

http://hihocoder.com/problemset/problem/1033
中问题意::
注意,是要求x的和,而不是x的数目。
数位dp。每一位进行一次计算(为了避免计算结果的式子长短不一。)
还有就是,求和的时候,要加上当前的影响,和下一位的结果。
根据位次求和,我觉得很好qwq

#include
using namespace std;
typedef long long ll;
/* 数位dp、
 注意,求得是x的和。
*/
const int maxn=1e3;
const ll mod=1e9+7;
struct Node{
   ll num,ans;
};
Node dp[20][500];
int bit[20];
ll base[20];
int yy;
Node dfs(int u,int all,int limit){
     Node s;
      s.ans=0;
      s.num=0;
     if(u==0){
        if(all==200)
           s.num=1;
        return s;//计数,计1
     }
     if(dp[u][all].num!=-1&&!limit)
        return dp[u][all];
     int ls=limit?bit[u]:9;
     int wq=(yy-u)%2?(-1):(1);
     int head;
     if(u==yy) head=1;//最高位不能为0.
     else head=0;
     s.num=0;
     s.ans=0;
     for(int i=head;i<=ls;i++){
         Node tmp=dfs(u-1,all-i*(wq),limit&&(i==bit[u]));
         if(tmp.num>0){
         s.num+=tmp.num%mod;
         s.ans+=((i*base[u]%mod)*tmp.num%mod)%mod;
         s.ans+=(tmp.ans)%mod;
         s.ans%=mod;
         }
     }
     if(!limit){
        dp[u][all]=s;
       // cout<
        }
     return s;
}
ll solve(ll x,int k){
     for(int i=0;i<20;i++)
        for(int j=0;j<500;j++)
        dp[i][j].num=-1;
     if(x<=0) return 0;
     int cc=0;
     int anss=0;
     while(x){
         bit[++cc]=x%10;
         x/=10;
         anss+=bit[cc];
     }
     ll sumss=0;
     //cout<
     for(yy=1;yy<=cc;yy++){
         //cout<
         memset(dp,-1,sizeof(dp));
         sumss=(sumss+dfs(yy,k+200,yy==cc).ans%mod)%mod;
        //cout<
      sumss=(sumss+mod)%mod;
    }
 //cout<
    return sumss;
}
int main()
{   ll l,r,k;
    base[1]=1;
    for(int i=2;i<=20;i++)
        base[i]=(base[i-1]*10)%mod;
    while(~scanf("%lld%lld%lld",&l,&r,&k)){
    for(int i=0;i<20;i++)
        for(int j=0;j<500;j++)
        dp[i][j].num=-1;
    printf("%lld\n",(solve(r,k)-solve(l-1,k)+mod)%mod);
    }

    return 0;
}

你可能感兴趣的:(数位统计dp)