数位dp——BZOJ1026 Windy数

1026: [SCOI2009]windy数

Time Limit: 1 Sec  Memory Limit: 162 MB

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 。

 

很经典的模板题。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define foru(i,x,y) for(int i=x;i<=y;i++)
#define clr(a) memset(a,0,sizeof(a))
using namespace std;
int bit[20],f[20][20],l,a,b;
int dfs(int pos,int pre,int lim){//当前位置 前一位的数字 是否有限制 
    if(pos<=0)return 1;
    if(pre>=0&&!lim&&f[pos][pre]!=-1)return f[pos][pre];//如果当前没有限制且此情况已搜索过，则返回 
    int rng=(lim?bit[pos]:9),ret=0;//确定范围 
    foru(i,0,rng)
        if(abs(i-pre)>=2)//如果符合要求 递归下一位 
            ret+=dfs(pos-1,(pre<0&&i==0)?pre:i,lim&&(i==rng));//下一位 判断是否为第一位 是否有限制 
    if(pre>0&&!lim)f[pos][pre]=ret;//若无限制 记录 
    return ret;
}

int calc(int k){
    int l=0;
    while(k){bit[++l]=k%10;k/=10;}
    memset(f,-1,sizeof(f));
    return dfs(l,-10,1);
}

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