SCOI2009 windy数

#include <cstdio>
#include <algorithm>
using namespace std;
const long long int p[15]={0,1,10,100,1000,10000,100000,1000000,10000000,100000000,1000000000,10000000000};
const int inf=0x3f3f3f;
int A,B,f[15][10];
void dp()
{
int i,j,k;
for(i=0;i<=9;i++)    //i=0
f[1][i]=1;         //f[1][i]=1;
for(i=2;i<=9;i++)    //i<=9
for(j=0;j<=9;j++)  //j=0
for(k=0;k<=9;k++)
if(abs(j-k)>=2)
f[i][j]+=f[i-1][k];
for(j=1;j<=2;j++)
for(k=0;k<=9;k++)
if(abs(j-k)>=2)
f[10][j]+=f[9][k];
return;
}
int work(int n)
{
int i,j,maxn=0,last,lastn,now;
int ans=0;
while(n>=p[maxn]) maxn++;
maxn--;
//处理位数小于n
for(i=1;i<maxn;i++)
for(j=1;j<=9;j++)
ans+=f[i][j];
//处理位数相同
last=inf;
lastn=n;
for(i=maxn;i>=1;i--)
{
now=lastn/p[i];
if(maxn==i) j=1;
else j=0;
for(;j<now;j++)                      //j=0;
if(abs(last-j)>=2) ans+=f[i][j];
if(abs(now-last)<2) break;
lastn%=p[i];
last=now;
}
return ans;
}
int main()
{
scanf("%d%d",&A,&B);
dp();
printf("%d",work(B+1)-work(A));
return 0;
}