bzoj1026 [SCOI2009]windy数

#include<bits/stdc++.h>
using namespace std;
int dp[15][10][2];
int digit[15];
int m,n;
int len;
int DFS(int pos,int pre,int limit,int flag)
{
if(pos==0)return 1;
if(!limit&&dp[pos][pre][flag]!=-1)return dp[pos][pre][flag];
int up=limit?digit[pos]:9;
int tmp=0;
for(int i=0;i<=up;i++)
{
if(abs(pre-i)>=2||pos==len||flag)tmp+=DFS(pos-1,i,limit&&i==up,flag&&i==0);
}
if(!limit)dp[pos][pre][flag]=tmp;
return tmp;
}
int solve(int x)
{
len=0;
while(x)digit[++len]=x%10,x/=10;
return DFS(len,0,1,1);
}
int main()
{
memset(dp,-1,sizeof(dp));
scanf("%d%d",&m,&n);
printf("%d",solve(n)-solve(m-1));
return 0;
}