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

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=1026

蛮简单的数位DP，预处理 f[i][j] 表示 i 位数，以 j 开头的 windy 数个数；

但不明白为什么最后一位拿出来特判 ret++  不对，而写在循环里，特判 i==1 就对了...

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a,b,f[][],num[];
int abb(int x){return (x>)?x:-x;}
int getnum(int x)
{
    int cnt=;
    while(x)num[++cnt]=x%,x/=;
    return cnt;
}
void init()
{
    int mx=getnum(b);
    for(int i=;i<=;i++)f[][i]=;
    for(int i=;i<=mx;i++)
        for(int j=;j<=;j++)
            for(int k=;k<=;k++)
                if(abb(j-k)>=)f[i][j]+=f[i-][k];
}
int calc(int x)
{
    int mx=getnum(x),ret=;
    for(int i=;i<mx;i++)
        for(int j=;j<=;j++)ret+=f[i][j];
    for(int j=;j<num[mx];j++)ret+=f[mx][j];
    for(int i=mx-,pre;i;i--)
    {
        pre=num[i+];
        if(i!=)
        {
            for(int j=;j<num[i];j++)
                if(abb(pre-j)>=)ret+=f[i][j];
        }
        if(i==)//AC
        {
            for(int j=;j<=num[i];j++)
                if(abb(pre-j)>=)ret+=f[i][j];
        }
        if(abb(num[i]-pre)<)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-));
    return ;
}