BZOJ 1026: [SCOI2009]windy数 数位DP

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#define ll long long
using namespace std;
ll a,b;
ll f[30][30],d[30],bin[30];
ll solve(ll x)
{
if(x==0) return 0;
memset(d,0,sizeof(d));
ll cnt1=0,res=0;
while(x){d[++cnt1]=x%10;x/=10;}
for(int i=1;i<d[cnt1];i++) res+=f[cnt1][i];
for(int i=cnt1-1;i>=1;i--)
{
for(int j=1;j<=9;j++)
{
res+=f[i][j];
}
}
for(int i=cnt1-1;i>=1;i--)
{
for(int j=0;j<d[i];j++)
if(abs(d[i+1]-j)>=2) res+=f[i][j];
if(abs(d[i+1]-d[i])<2) break;
}
return res;
}
int main()
{
scanf("%lld%lld",&a,&b);
bin[1]=1;
for(int i=2;i<=10;i++) bin[i]=bin[i-1]*10;
for(int i=0;i<=9;i++) f[1][i]=1;
for(int i=2;i<=10;i++)
{
for(int j=0;j<=9;j++)
{
for(int k=0;k<=9;k++)
{
if(abs(j-k)>=2) f[i][j]+=f[i-1][k];
}
}
}
printf("%lld",solve(b+1)-solve(a));
return 0;
}