UESTC 1307 windy数(数位DP)

在区间内统计任意两位相差大于二的数的个数。数位dp在推到时候注意最高位要特判一下因为最高位是没有前一位的。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#define ll long long
#define LEN 100

using namespace std;

ll dp[LEN][LEN];

inline int abs(int x){return x>0?x:-x;}

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

ll solve(ll x){
    if(!x)return 0;
    ll ret = 0;
    int str[LEN] = {0};
    int len = 0;
    while(x){
        str[++len] = x%10;
        x/=10;
    }
    bool flag = true;
    for(int i=1; i<len; i++){
        for(int j=1; j<=9; j++){
            ret+=dp[i][j];
        }
    }
    for(int i=len; i>=1; i--){
        for(int j=0; j<str[i]; j++){
            if(i==len && !j) continue;
            if(abs(str[i+1]-j)<2 && i!=len)continue;
            ret += dp[i][j];
        }
        if(abs((str[i+1])-(str[i]))<2 && i!=len){
            flag = false;
            break;
        }
    }
    if(flag)ret++;
    return ret;
}

int main()
{
//    freopen("in.txt", "r", stdin);
    ll a, b;
    init();
    while(cin >> a >> b){
        ll ans = solve(b)-solve(a-1);
        cout << ans << endl;
    }
    return 0;
}