BZOJ 1833 数位DP 解题报告

1833: [ZJOI2010]count 数字计数

Description

给定两个正整数a和b，求在[a,b]中的所有整数中，每个数码(digit)各出现了多少次。

Input

输入文件中仅包含一行两个整数a、b，含义如上所述。

Output

输出文件中包含一行10个整数，分别表示0-9在[a,b]中出现了多少次。

Sample Input

1 99

Sample Output

9 20 20 20 20 20 20 20 20 20

HINT

30%的数据中，a<=b<=10^6；
100%的数据中，a<=b<=10^12。

代码如下：（谁能告诉我为什么把主函数里的i定义到for里面就WA了）

#include<cstdio>  #include<cstring>  #include<iostream>  #include<algorithm>  using namespace std;  long long ans[10],f[20];  inline void Resolve(long long x,long long pos)  {      while(x)          ans[x%10]+=pos,x/=10;  }  void Digital_DP(long long x,int flag)  {      int i,j;      long long pos,now;      for(i=1,pos=10;pos<x;++i,pos*=10)      {          for(j=0;j<=9;j++)              ans[j]+=f[i-1]*9*flag;          for(j=1;j<=9;j++)              ans[j]+=pos/10*flag;      }      now=pos/=10;--i;      while(now<x)      {          while(now+pos<=x)          {              long long temp=now/pos;              Resolve(temp,pos*flag);              for(j=0;j<=9;j++)                  ans[j]+=f[i]*flag;              now+=pos;          }          pos/=10;--i;      }  }  int main()  {        int i,j;    long long a,b,pos;      f[1]=1;     for(i=2,pos=10;i<=12;i++,pos*=10)          f[i]=f[i-1]*10+pos;      scanf("%lld%lld",&a,&b);    Digital_DP(b+1,1);      Digital_DP(a,-1);     for(j=0;j<=9;j++)          printf("%lld%c",ans[j],j==9?'\n':' ');      return 0;}