windy数（数位DP）



Description

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，在A和B之间，包括A和B，总共有多少个windy数？

Input

包含两个整数，A B。

Output

一个整数。

Sample Input

【输入样例一】
1 10
【输入样例二】
25 50

Sample Output

【输出样例一】
9
【输出样例二】
20

Hint

【数据规模和约定】

100%的数据，满足 1 <= A <= B <= 2000000000 。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lson l,m,rt <<1
#define rson m+1,r,rt << 1 | 1
#define LL long long
#define INF 0x3f3f3f3f
int dir4[4][2]= {{0,1},{1,0},{0,-1},{-1,0}};
int dir8[8][2]= {{0,1},{1,1},{1,0},{1,-1},{0,-1},{-1,-1},{-1,0},{-1,1}};
int dir_8[8][2]= {{-2,1},{-1,2},{1,2},{2,1},{2,-1},{1,-2},{-1,-2},{-2,-1}};
using namespace std;
int dp[20][10]; //dp[i][j]表示考虑i位的数中，最高为j的windy数
int abs(int x)
{
    return x<0?-x:x;
}

void Init()
{
    memset(dp,0,sizeof(dp));
    for(int i=0; i<=9; i++)
        dp[1][i]=1;
    for(int i=2; i<=10; i++)
        for(int j=0; j<10; j++)
            for(int k=0; k<10; k++)
                if(abs(j-k)>=2)
                    dp[i][j]+=dp[i-1][k];
}

int Solve(int x)
{
    int digit[20],cnt=0;
    while(x)
    {
        digit[++cnt]=x%10;
        x/=10;
    }
    digit[cnt+1]=0;
    int ans=0;
    for(int i=1; i<cnt; i++) //先把长度为1至cnt-1计入
        for(int j=1; j<10; j++)
            ans+=dp[i][j];
    for(int j=1; j<digit[cnt]; j++) //确定最高位
        ans+=dp[cnt][j];
    for(int i=cnt-1; i>0; i--)
    {
        for(int j=0; j<digit[i]; j++)
            if(abs(j-digit[i+1])>=2)
                ans+=dp[i][j];
        if(abs(digit[i]-digit[i+1])<2)  //如果高位已经出现非法，直接退出
            break;
    }
    return ans;
}

int main()
{
    int a,b;
    Init();
    while(~scanf("%d%d",&a,&b))
    {
        printf("%d\n",Solve(b+1)-Solve(a));
    }
    return 0;
}