【BZOJ 1026】 [SCOI2009]windy数

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 。

 

数位DP，f[i][j]表示一共有i位最高位为j共有多少个符合条件的值

 #include<cstdio>
 #include<cmath>
 using namespace std;
 int base[],f[][],a,b;
 void pre(){
     base[]=;
     for(int i=;i<=;i++) base[i]=base[i-]*;//base代表这一位有多少数字
     for(int i=;i<=;i++)f[][i]=;//f[i][j]代表有i位第一位为j的有多少个
     for(int i=;i<=;i++)
         for(int j=;j<=;j++)
             for(int k=;k<=;k++)
                 if(fabs(j-k)>=) f[i][j]+=f[i-][k]; 

 }

 int calc(int x){
     if(!x) return ;
     int w=,ans=;
     while(base[w]>x)w--;
     int cur=x/base[w];//x的第一位
     for(int i=;i<w;i++)
         for(int j=;j<=;j++) ans+=f[i][j];//why?
     for(int i=;i<cur;i++) ans+=f[w][i];
     x%=base[w];
     int pre=cur;
     for(int i=w-;i;i--){
         cur=x/base[i];
         if(i>){
             for(int j=;j<cur;j++)if(fabs(j-pre)>=) ans+=f[i][j];
         }
         else{
             for(int j=;j<=cur;j++) if(fabs(j-pre)>=) ans+=f[i][j];
         }
         if(fabs(cur-pre)<) break;
         x%=base[i];
         pre=cur;
     }
     return ans;
 }

 int main(){
     scanf("%d%d",&a,&b);
     pre();
     printf("%d",calc(b)-calc(a-));
 }