BZOJ 1026 数位DP

数位DP

Description

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

Input

　　包含两个整数，A B。

Output

　　一个整数

思路：

​ 这一类的简单数位DP，可以在限制条件下，处理所有的可能。

​ 然后在细致考虑最大的哪一种情况，满足条件下求和即可。

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

typedef long long LL;

LL l,r;
LL dp[20][20];
int bit[20];

int abs(int x)
{
    if(x < 0) return -x;
    return x;
}

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

LL solve(LL x)
{
    int pos = 0;
    while(x) {
        bit[++pos] = x%10;
        x /= 10;
    }
    bit[pos+1] = 0;
    long long ans = 0;
    for(int i = 1;i < pos; i++)
        for(int j = 1;j <= 9; j++)
            ans += dp[i][j];
    for(int i = 1;i < bit[pos]; i++)
        ans += dp[pos][i];
    for(int i = pos-1;i >= 1; i--) {
        for(int j = 0;j < bit[i]; j++)
            if(abs(j-bit[i+1]) >= 2)
                ans += dp[i][j];
        if(abs(bit[i+1]-bit[i]) < 2) break;
    }
    return ans;
}

int main(int argc, char const *argv[])
{
    // freopen("in.txt","r",stdin);
    init();
    while(scanf("%lld%lld",&l,&r) != EOF) {
        printf("%lld\n",solve(r+1)-solve(l));
    }

    return 0;
}