[Jzoj]1039.【SCOI2009】windy数

题目大意

$windy$定义了一种$windy$数。
不含前导零且相邻两个数字之差至少为$2$的正整数被称为$windy$数。
$windy$想知道，在$A$和$B$之间，包括$A$和$B$，总共有多少个$windy$数？

题目解析

设 $ans[n]$为 $0$-$n$ 中$windy$ 数的个数，求 $A$-$B$ 以内的 $windy$ 数个数，也就是求$ans[B]-ans[A-1]$

很明显是数位$DP$，很经典的题。

代码

#include<bits/stdc++.h>
using namespace std;
int f[15][15][2][2],a[15];
int dp(int pos,int pre,bool zero,bool lim)
{
	if(pos<1) return 1;
	if(f[pos][pre][zero][lim]>=0) return f[pos][pre][zero][lim];
	int end=lim?a[pos]:9,ans=0;
	for(int i=0;i<=end;i++)
	 if(abs(i-pre)>=2||zero)
	  ans+=dp(pos-1,i,zero&&(i==0),lim&&(i==end));
	f[pos][pre][zero][lim]=ans;
	return ans;
}
int calc(int x)
{
	int cnt=0;
	memset(f,-1,sizeof(f));
	memset(a,0,sizeof(a));
	while(x) a[++cnt]=x%10,x/=10;
	dp(cnt,0,1,1);
}
int main()
{
	int a,b;
	cin>>a>>b;
	cout<<calc(b)-calc(a-1);
}