数位dp——BZOJ1026 Windy数

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #define foru(i,x,y) for(int i=x;i<=y;i++)
 5 #define clr(a) memset(a,0,sizeof(a))
 6 using namespace std;
 7 int bit[20],f[20][20],l,a,b;
 8 int dfs(int pos,int pre,int lim){//当前位置 前一位的数字 是否有限制 
 9     if(pos<=0)return 1;
10     if(pre>=0&&!lim&&f[pos][pre]!=-1)return f[pos][pre];//如果当前没有限制且此情况已搜索过，则返回 
11     int rng=(lim?bit[pos]:9),ret=0;//确定范围 
12     foru(i,0,rng)
13         if(abs(i-pre)>=2)//如果符合要求 递归下一位 
14             ret+=dfs(pos-1,(pre<0&&i==0)?pre:i,lim&&(i==rng));//下一位 判断是否为第一位 是否有限制 
15     if(pre>0&&!lim)f[pos][pre]=ret;//若无限制 记录 
16     return ret;
17 }
18 
19 int calc(int k){
20     int l=0;
21     while(k){bit[++l]=k%10;k/=10;}
22     memset(f,-1,sizeof(f));
23     return dfs(l,-10,1);
24 }
25 
26 int main(){
27     scanf("%d%d",&a,&b);
28     printf("%d\n",calc(b)-calc(a-1));
29 }