【SCOI 2009】 Windy数

【题目链接】

            点击打开链接

【算法】

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

【代码】

           

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

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

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

int main() {
    
        getdp();
        read(n); read(m);
        writeln(calc(m+1)-calc(n));
        
        return 0;
}