BZOJ 1026: [SCOI2009]windy数 数位DP

Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 8370 Solved: 3770

Description

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

Input

　　包含两个整数，A B。

Output

　　一个整数

Sample Input

【输入样例一】

1 10

【输入样例二】

25 50
Sample Output

【输出样例一】

9

【输出样例二】

20
HINT

【数据规模和约定】

100%的数据，满足 1 <= A <= B <= 2000000000 。

Source

---

数位DP裸题，按照套路来就可以了

#include <bits/stdc++.h> 
using namespace std;
int f[15][15], base[15];
int A, B;
void init( ) {
    base[1] = 1;
    for( register int i = 2; i <= 10; i++ ) base[i] = base[ i - 1 ] * 10;
    scanf( "%d%d", &A, &B );
    for( register int i = 0; i <= 9; i++ ) f[1][i] = 1;
    for( register int i = 2; i <= 10; i++ )//枚举多少位数 
        for( register int j = 0; j <= 9; j++ )//枚举这一位是什么 
            for( register int k = 0; k <= 9; k++ ) //最高位是j的情况下,次高位可以是k 
                if( abs( j - k ) >= 2 )f[i][j] += f[ i - 1 ][k];
}
int solve( int n ) {
    if( n == 0 ) return 0;
    int tmp = 0, maxn = 10;
    while( base[maxn] > n ) maxn--;
    for( register int i = 1; i < maxn; i++ )
        for( register int j = 1; j <= 9; j++ )
            tmp += f[i][j];
    int hi = n / base[maxn];
    for( register int i = 1; i < hi; i++ ) tmp += f[maxn][i];
    n %= base[maxn];
    int prehi = hi;
    for( register int i = maxn - 1; i; i-- ) {
        hi = n / base[i];
        if( i != 1 ) {
            for( register int j = 0; j < hi; j++ ) if( abs( prehi - j ) >= 2 ) tmp += f[i][j];
        } else {
            for( register int j = 0; j <= hi; j++ ) if( abs( prehi - j ) >= 2 ) tmp += f[i][j];
        }
        if( abs( hi - prehi ) < 2 ) break;
        prehi = hi;
        n %= base[i];
    }
    return tmp;
} 
int main( ) {
    init( );
    printf( "%d", solve(B) - solve( A - 1 ) );
    return 0;
}