bzoj 1026 [ SCOI2009 ] windy数 —— 数位DP

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a,b,f[15][15],num[15];
int abb(int x){return (x>0)?x:-x;}
int getnum(int x)
{
    int cnt=0;
    while(x)num[++cnt]=x%10,x/=10;
    return cnt;
}
void init()
{
    int mx=getnum(b);
    for(int i=0;i<=9;i++)f[1][i]=1;
    for(int i=2;i<=mx;i++)
        for(int j=0;j<=9;j++)
            for(int k=0;k<=9;k++)    
                if(abb(j-k)>=2)f[i][j]+=f[i-1][k];
}
int calc(int x)
{
    int mx=getnum(x),ret=0;
    for(int i=1;i<mx;i++)
        for(int j=1;j<=9;j++)ret+=f[i][j];
    for(int j=1;j<num[mx];j++)ret+=f[mx][j];
    for(int i=mx-1,pre;i;i--)
    {
        pre=num[i+1];
        if(i!=1)
        {
            for(int j=0;j<num[i];j++)
                if(abb(pre-j)>=2)ret+=f[i][j];
        }
        if(i==1)//AC
        {
            for(int j=0;j<=num[i];j++)
                if(abb(pre-j)>=2)ret+=f[i][j];
        }
        if(abb(num[i]-pre)<2)break;//!
    }
//    if(mx==1||abb(num[1]-num[2])>=2)ret++;//WA
//    if(abb(num[1]-num[2])>=2)ret++;//WA
    return ret;
}
int main()
{
    scanf("%d%d",&a,&b);
    init();
    printf("%d\n",calc(b)-calc(a-1));
    return 0;
}