uestc 1307 windy数 数位DP

主要注意下0的情况就可以！！！

链接：http://acm.uestc.edu.cn/problem.php?pid=1307

代码如下：

 

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll __int64
using namespace std;
int bit[10],dp[20][20];
int dfs(int pos,int pre,bool h,bool f)
{
    if(pos==-1) return h==0;
    if(!f&&dp[pos][pre]!=-1&&!h) return dp[pos][pre];
    int ans=0;
    int e=f?bit[pos]:9;
    for(int i=0;i<=e;i++){
        if(h) ans+=dfs(pos-1,i,h&&i==0,f&&(i==bit[pos]));
        else if(abs(pre-i)>=2)ans+=dfs(pos-1,i,h,f&&(i==bit[pos]));
    }
    if(!f&&!h) dp[pos][pre]=ans;
    return ans;
}
int cal(int n)
{
    int m=0;
    while(n){
        bit[m++]=n%10;
        n/=10;
    }
    return dfs(m-1,0,1,1);
}
int main()
{
    int n,m;
    memset(dp,-1,sizeof(dp));
    while(scanf("%d%d",&n,&m)!=EOF){
        printf("%d\n",cal(m)-cal(n-1));
    }
}