Windy数 数位dp

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#define IL inline
#define x first
#define y second
typedef long long ll;
using namespace std;
const	int N=11;
int  f[N][N];// f[i][j]代表长度为i最高位为1的方案数
void init()
{
	for(int i=0;i<=9;i++)	f[1][i]=1;
	for(int i=2;i<N;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];
}
int dp(int n)
{
	if(!n)	return 0;		// 0不是windy数
	vector<int>v;
	while(n)	v.push_back(n%10),n/=10;
	int last=-2,res=0; 		//把last设置成每个数减去它都大于等于2的数字
	for(int i=v.size()-1;i>=0;i--)
	{
		int   x=v[i];
		for(int j=( i==v.size()-1);j<x;j++)
		if(abs(j-last)>=2)	res += f[i+1][j];
		if(abs(x-last)<2)	break;
		else	last=x;			//last代表Ak-1
		if(!i)	res++;		//判断右边分支
	}
	for(int i=1;i<=v.size()-1;i++)
	for(int j= 1;j<=9;j++)
	res+=f[i][j];
	return res;
}
int main()
{
	init();
	int l,r;
	cin>>l>>r;
	cout<<dp(r)-dp(l-1)<<endl;
    return 0;
}