bzoj1026 [SCOI2009]windy数（数位dp）

dp[i][j]表示长度为i，最高位为j的windy数有多少个。则显然有
$dp[i][j]=\sum\limits_{k} dp[i-1][k]*[abs(k-j)>=2]$。
那么答案就是calc(b)-calc(a-1)。
如何计算1~x的windy数呢？
假设x有n位，从低到高第i位为a[i]。
则首先所有长度为1~n-1的windy数都要加进来。
所有长度为n，最高位< a[n]的都要加进来。
然后考虑第n位是a[n]的，则第n-1位要是0~a[n-1]-1，且与a[n]的差>=2，加上他的答案。然后确定第n-1位是a[n-1]，再枚举第n-2位，以此类推。如果某时a[i]与a[i+1]的差小于2了，说明已经不是windy数了，要直接退出。
我们发现，这样统计的答案其实是1~x-1的windy数，并没有判断x。为了不加特判，我们求的答案改成calc(a+1)-calc(b)就好了。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int dp[11][10];
inline int calc(int x){//实际上统计的是1~x-1的windy数的个数 
    int tot=0,a[20],ans=0;memset(a,0,sizeof(a));
    while(x) a[++tot]=x%10,x/=10;
    for(int i=1;i<tot;++i)
        for(int j=1;j<=9;++j) ans+=dp[i][j];
    for(int i=1;i<a[tot];++i) ans+=dp[tot][i];
    for(int i=tot-1;i>=1;--i){
        for(int j=0;j<a[i];++j){
            if(abs(j-a[i+1])<2) continue;
            ans+=dp[i][j];
        }if(abs(a[i+1]-a[i])<2) break;//已经不是windy数了，直接退出 
    }return ans;
}
int main(){
//  freopen("a.in","r",stdin);
    int a=read(),b=read();
    for(int i=0;i<=9;++i) dp[1][i]=1;
    for(int len=2;len<=10;++len)
        for(int i=0;i<=9;++i)
            for(int j=0;j<=9;++j){
                if(abs(i-j)<2) continue;
                dp[len][i]+=dp[len-1][j];
            }
    printf("%d\n",calc(b+1)-calc(a-1+1));
    return 0;
}