BZOJ1026 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 。

数位DP，f[i][j] 表示第i位取j的方案数。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int f[15][15]={0},x,nx=0,ny=0,y,a[15],b[15];
int main()
{
    while (scanf("%d%d",&x,&y)==2){
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    nx=0;ny=0;
    for (int i=0;i<=9;i++)
        f[1][i]=1;
    for (int i=2;i<=11;i++)
        for (int j=0;j<=9;j++)
            for (int k=0;k<=9;k++)
                if (fabs(k-j)>=2)
                    f[i][j]+=f[i-1][k];
    int ans1=0,ans2=0;
    while (x)
        {
            nx++;
            a[nx]=x%10;
            x/=10;
        }
    a[nx+1]=0;
    for (int i=1;i<nx;i++)
        for (int j=1;j<=9;j++)
            ans1+=f[i][j];
    for (int i=1;i<a[nx];i++)
        ans1+=f[nx][i];
    for (int i=nx-1;i>0;i--)
        {
            for (int j=0;j<a[i];j++)
                if (fabs(a[i+1]-j)>=2)
                    ans1+=f[i][j];
            if (fabs(a[i+1]-a[i])<=1)
                break;
        }
    y++;
    while (y)
        {
            ny++;
            b[ny]=y%10;
            y/=10;
        }
    b[ny+1]=0;
    for (int i=1;i<ny;i++)
        for (int j=1;j<=9;j++)
            ans2+=f[i][j];
    for (int i=1;i<b[ny];i++)
        ans2+=f[ny][i];
    for (int i=ny-1;i>0;i--)
        {
            for (int j=0;j<b[i];j++)
                if (fabs(b[i+1]-j)>=2)
                    ans2+=f[i][j];
            if (fabs(b[i+1]-b[i])<=1)
                break;
        }
    printf("%d\n",ans2-ans1);
    }
    return 0;
}