【BZOJ】1026 [SCOI2009] windy数

Description

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

Input

　　包含两个整数，A B。

Output

　　一个整数

Solution

数位DP，考虑一下压上界的边界情况的处理记录就好。
另一个可能卡人的地方是多个前导零。

#include<stdio.h>
#include<algorithm>
#include<cstring>

using namespace std;

typedef long long ll;

ll f[20][10][2],a,b;
int lim[20];

inline ll solve(ll a)
{
    int len=0;
    for (;a;lim[++len]=a%10,a/=10);
    memset(f,0,sizeof(f));
    for (int i=1;i<lim[len];i++) f[len][i][0]=1;
    f[len][lim[len]][1]=1;
    for (int i=len-1;0<i;i--) for (int j=0;j<10;j++) for (int k=0;k<10;k++) if (j+1<k || k+1<j)
    {
        f[i][j][0]+=f[i+1][k][0];
        if (j<lim[i]) f[i][j][0]+=f[i+1][k][1];
        if (j==lim[i]) f[i][j][1]+=f[i+1][k][1];
    }
    ll rt=0;
    for (int i=0;i<10;i++) rt+=f[1][i][0]+f[1][i][1];
    for (int up=len-1;0<up;up--)
    {
        memset(f,0,sizeof(f));
        for (int i=1;i<10;i++) f[up][i][0]=1;
        for (int i=up-1;0<i;i--) for (int j=0;j<10;j++) for (int k=0;k<10;k++) if (j+1<k || k+1<j) f[i][j][0]+=f[i+1][k][0];
        for (int i=0;i<10;i++) rt+=f[1][i][0];
    }
    return rt;
}

int main()
{
    scanf("%lld%lld",&a,&b);
    if (b<a) swap(a,b);
    printf("%lld",solve(b)-solve(a-1));
}