[SCOI2009] [BZOJ1026] Windy数 - 数位DP

1026: [SCOI2009]windy数

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 4858   Solved: 2198
[ Submit][ Status][ Discuss]

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，因为有前导0的关系，实现过程略麻烦。

#include "stdio.h"
using namespace std;
const int size=10,sizef=11;
int f[sizef][sizef];

int getAnsw(int t){
    int wd[sizef]={0},l=0;
    int k=t;
    while (t) wd[++l]=t%10,t/=10;
    int ans=0,i,j;
    if(l==1) return k;
    for (i=1;i<l;i++)
     for (j=1;j<size;j++)
      ans+=f[j][i];
    for (i=1;i<wd[l];i++)
     ans+=f[i][l];
    for (j=l-1;j>0;j--){
        for (i=0;i<wd[j]||(j==1&&i<=wd[j]);i++){ 
            if(i>=wd[j+1]+2||i<=wd[j+1]-2) ans+=f[i][j];
        }
        if(wd[j]<wd[j+1]+2&&wd[j]>wd[j+1]-2) break;
    }
    return ans;
}
int main(){
    int i,j,k;
    for (i=0;i<sizef;i++)
     f[i][1]=1;
    for (j=2;j<sizef;j++){
     for (i=0;i<size;i++){
      for (k=0;k<size;k++)
       if(k>=i+2||k<=i-2) f[i][j]+=f[k][j-1];
     }
    }
    int a,b;
    scanf("%d%d",&a,&b);
    printf("%d\n",getAnsw(b)-getAnsw(a-1));
    return 0;
}