数位dp bzoj 1026 windy数

bzoj 1026 windy数

2014年6月18日3,0053
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
【数据规模和约定】
20%的数据，满足 1 <= A <= B <= 1000000 。
100%的数据，满足 1 <= A <= B <= 2000000000 。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define LL long long
using namespace std;

int f[15][15];//f[i][j]代表i位数，首位为j，的个数
int L, R, sum, num;

int work(int n){
    int cc = n; sum = 0; num = 1;
    if( !n ) return 0;
    while(cc /= 10) num++;//位数 
    for(int i=1; i<num; i++)//不考虑同位数 
        for(int j=1; j<=9; j++)//完全计数
            sum += f[i][j];
    int head = n / pow(10,num-1);//首位 
    for(int i=1; i<head; i++)//分块处理 
        sum += f[num][i];
    n %= (int)pow(10,num-1);
    int now = head;
    for(int i=num-1; i; i--){//i代表当前位数 
        head = n / pow(10,i-1);
        if(i != 1){
            for(int j=0; j<head; j++)
                if(abs(now - j) >= 2) sum += f[i][j];
        }
        else {
            for(int j=0; j<=head; j++)//特判！！
               if(abs(now - j) >= 2) sum += f[i][j];
        }
        if(abs(head - now) < 2) break;
        now = head;
        n %= (int)pow(10,i-1);//相同操作 
    }
    return sum;
}

int main(){
    scanf("%d%d", &L, &R);
    for(int i=0; i<=9; i++) 
        f[1][i] = 1;
    for(int i=2; i<=11; i++)
        for(int j=0; j<=9; j++)
            for(int k=0; k<=9; k++)
                if(abs(j - k) >= 2) 
                    f[i][j] += f[i-1][k];//初始化不考虑排除前导零（之后会用到） 
    printf("%d", work(R) - work(L-1));
    return 0;
}