[SCOI2009]windy数 数位dp

原题:https://www.luogu.org/problemnew/show/P2657

题解:求区间不含前导0相邻两个数位之差至少为2的数。将区间拆开，考虑数位dp设f[i][j]为剩i位，当前位为j的方案数。显然f[i][j]=sum({[i-1][k]}(|k-j|>=2) 。用记忆化搜索就行了。

细节:先导0要注意，可以将j设为11表示最高位为0的情况。

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
vector<int> dim;
ll l,r;
ll f[30][30];
ll dfs(int x,int p,int lim){
	if(!x) return 1;
	if(!lim && f[x][p]!=-1) return f[x][p];
	int maxm=(lim?dim[x]:9);
	ll ans=0;
	for(int i=0;i<=maxm;i++){
		if(abs(i-p)<2) continue;
		if(p==11 && i==0) ans+=dfs(x-1,11,lim&(i==maxm));
		else ans+=dfs(x-1,i,lim&(i==maxm));
	} 
	if(!lim) f[x][p]=ans; 
	
	return ans;
} 
ll work(ll x){
	dim.clear();dim.push_back(-1);
	ll t=x;
	while(t) dim.push_back(t%10),t/=10;
	return dfs(dim.size()-1,11,1); 
}
int main(){
	scanf("%lld%lld",&l,&r);
	memset(f,-1,sizeof f);
	printf("%lld\n",work(r)-work(l-1));
	return 0;
}