uestc 250 windy数（数位dp）

题意：不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。

windy想知道，在A和B之间，包括A和B，总共有多少个windy数？

思路：数位dp

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

int dp[][];//dp[i][j]表示长度为i，最高位为j的windy数的个数

void init(){
    memset(dp,,sizeof(dp));
    int i,j,k;
    for(j=;j<;++j)dp[][j]=;
    for(i=;i<=;++i)
        for(j=;j<=;++j){
            for(k=;k<=j-;++k)dp[i][j]+=dp[i-][k];
            for(k=j+;k<=;++k)dp[i][j]+=dp[i-][k];
        }
}

int bit[];
int calc(int n){
    if(n==)return ;
    int len=,i,j;
    while(n){
        bit[++len]=n%;
        n/=;
    }
    bit[len+]=-;
    bool flag=true;
    int ans=;
    for(i=;i<len;++i)//先把长度小于len的计入
        for(j=;j<=;++j)
            ans+=dp[i][j];
    for(j=;j<bit[len];++j)ans+=dp[len][j];//最高位
    for(i=len-;i>=;--i){
        for(j=;j<bit[i];++j)
            if(abs(bit[i+]-j)>=)
                ans+=dp[i][j];
        if(abs(bit[i+]-bit[i])<){
            flag=false;
            break;
        }
    }
    if(flag)++ans;//加上n本身
    return ans;
}

int main(){
    init();
    int a,b;
    while(~scanf("%d%d",&a,&b)){
        printf("%d\n",calc(b)-calc(a-));
    }
    return ;
}