【SCOI 2009】 Windy数

【题目链接】

点击打开链接

【算法】

数位DP，注意处理前导零的情况

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXL 15

int n,m;
int dp[MAXL][],a[MAXL];

template <typename T> inline void read(T &x) {
        int f = ; x = ;
        char c = getchar();
        for (; !isdigit(c); c = getchar()) { if (c == '-') f = -f; }
        for (; isdigit(c); c = getchar()) x = (x << ) + (x << ) + c - '';
        x *= f;
}
template <typename T> inline void write(T x) {
        if (x < ) { x = -x; putchar('-'); }
        if (x > ) write(x/);
        putchar(x%+'');
}
template <typename T> inline void writeln(T x) {
        write(x);
        puts("");
}
inline void getdp() {
        int i,j,k;
        for (i = ; i <= ; i++) dp[][i] = ;
        for (i = ; i <= MAXL; i++) {
                for (j = ; j <= ; j++) {
                        for (k = ; k <= ; k++) {
                                if (abs(j-k) >= ) dp[i][j] += dp[i-][k];
                        }
                }
        }
}
inline int calc(int n) {
        int i,j,ans = ;
        a[] = ;
        while (n) {
                a[++a[]] = n % ;
                n /= ;
        }
        for (i = ; i < a[]; i++) {
                for (j = ; j <= ; j++) {
                        ans += dp[i][j];
                }
        }
        for (i = a[]; i >= ; i--) {
                for (j = ; j < a[i]; j++) {
                        if (i == a[] && !j) continue;
                        if ((i == a[]) || (abs(j-a[i+]) >= ))
                                ans += dp[i][j];
                }
                if (i != a[] && abs(a[i]-a[i+]) < ) break;
        }
        return ans;
}

int main() {

        getdp();
        read(n); read(m);
        writeln(calc(m+)-calc(n));

        return ;
}