P2657 [SCOI2009] windy 数 数位DP基础题

经典数位dp

记录的状态：

if0，上一位是否是前导0；

limit，上一位是否达到上界；

p，上一位的取值；

按照题意把不符合条件的剔除+记忆化+处理好前导0即可。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
const double PI= acos(-1.0);
const int M = 1e5+7;

int di[32];
int dp[32][10];//第i+1位的数字为j，第i位及一下的数字任意，这样windy数有多少 
//当前位len，上一位是否是前导0，是否达到限制，值为p，的限制下，windy数的个数是多少 
int dfs(int len,bool if0,bool limit,int p)
{
	if(len==0)return 1;
	//上一位未达到上限，且不是前导0，则直接导入dp 
	if(!limit&&!if0&&dp[len][p])return dp[len][p];
	int cnt=0,up=limit?di[len]:9;
	for(int i=0;i<=up;i++)
	{
		if(!if0 && abs(i-p)<2)continue;
		if(i==0&&if0)cnt+=dfs(len-1,true,limit&&(i==up),i);
		else
			cnt+=dfs(len-1,false,limit&&(i==up),i);
	}
	if(!limit&&!if0)dp[len][p]=cnt;
	return cnt; 
}
ll cal(int n)
{
	int k=0;
	while(n)
	{
		di[++k]=n%10;
		n/=10;
	}
	dfs(k,true,true,0);
}
int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	int a,b;
  	cin>>a>>b;
  	cout<<cal(b)-cal(a-1)<<endl;
	return 0;
}