最近新学了数位DP,bzoj1833是不错的练手入门题。
这题难度适中,有人可能会被a和b较大的值吓到,其实用数位DP是容易的。
用f[i][j][k]表示到有i位数,首位为j,数码k的个数。
可以先初始化好一张这样的表,然后读入时操作一下就好了,自己yy一下就好。
代码:
/**************************************************************
Problem: 1833
User: fantasticwtl
Language: C++
Result: Accepted
Time:0 ms
Memory:1300 kb
****************************************************************/
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
unsigned long long a,b,f[21][10][10];
unsigned long long sol (unsigned long long x,int y)
{
int len=0,wz[21]= {0};
unsigned long long ans=0,g=x-1;
while(x)wz[++len]=x%10,x/=10;
for(int i=1; i<len; i++)
for(int j=1; j<=9; j++)ans+=f[i][j][y];
for(int j=1; j<wz[len]; j++)ans+=f[len][j][y];
for(int i=len-1; i>=1; i--) {
for(int j=0; j<wz[i]; j++)ans+=f[i][j][y];
if(wz[i+1]==y)ans+=g-(1ll*pow(10,i)*y)+1;
g-=1ll*pow(10,i)*wz[i+1];
}
return ans;
}
int main ()
{
for(int i=0; i<=9; i++)f[1][i][i]=1;
for(int i=2; i<=12; i++)
for(int j=0; j<=9; j++) {
for(int kk=0; kk<=9; kk++)
for(int k=0; k<=9; k++)f[i][j][kk]+=f[i-1][k][kk];
f[i][j][j]+=pow(10,i-1);
}//初始化f数组
cin>>a>>b;//读入
for(int i=0; i<=8; i++)cout<<(sol(b+1,i)-sol(a,i))<<' ';
cout<<(sol(b+1,9)-sol(a,9))<<endl;//行末不能有空格,会PE
}