【数位dp】 P2657 [SCOI2009] windy 数

数位dp：

dp[i][op][v]表示第i位上的数字为v，并且状态是op

所以小于等于n的满足条件的数量为ans=dp[位数][2][0]+dp[位数][0][0-最高位上的数字-1]+dp[位数][1][最高位上的数字]

其中op=0表示之前的数字没有和所给数字完全一样，也就是没有到上限，因此，这一位选0-9都可以，且op=0

op=1表示之前的数字和所给数字完全一样，也就是达到上香，这一位只能选0-这一位，当选原数字上的这个数时，op=1

当选其他数字时，op就可以为0了

op=2 表示前面全是先导0（如：00012），注意这里存在00000的情况，但由于后续的相减，减掉了，不用做特别处理

我们就需要用ansb-ans（a-1）即为答案

#include<bits/stdc++.h>
using namespace std;
int a,b,w[11];
int dp[11][3][11];
int dpp(int i ,int op ,int v)
{
	if(dp[i][op][v]!=-1) return dp[i][op][v];
	if(i==1) return 1;
	if(i==0) return 0;
	dp[i][op][v]=0;
	if(op==0)
	{
		for(int j=0;j<=9;j++)
			if(v-j>=2 || j-v>=2)
				dp[i][op][v]+=dpp(i-1,op,j);
		return dp[i][op][v];
	}
	if(op==1)
	{
		for(int j=0;j<w[i-1];j++)
			if(v-j>=2 || j-v>=2)
				dp[i][op][v]+=dpp(i-1,0,j);
		if(w[i-1]-v>=2 || v-w[i-1]>=2)
			dp[i][op][v]+=dpp(i-1,1,w[i-1]);
		return dp[i][op][v];
	}
	if(op==2)
	{
		for(int j=1;j<=9;j++)
			dp[i][op][v]+=dpp(i-1,0,j);
		dp[i][op][v]+=dpp(i-1,2,0);
		return dp[i][op][v];
	}
}
int main()
{
	scanf("%d%d",&a,&b);
	int cnt=0,aa=b,bb=a-1;
	while(aa>0)
	{
		w[++cnt]=aa%10;
		aa/=10;
	}	
	memset(dp,-1,sizeof(dp));
	int ansa=0;
	ansa+=dpp(cnt,2,0);
	for(int i=1;i<w[cnt];i++)
		ansa+=dpp(cnt,0,i);
	ansa+=dpp(cnt,1,w[cnt]);
	cnt=0;
	memset(dp,-1,sizeof(dp));
	while(bb>0)
	{
		w[++cnt]=bb%10;
		bb/=10;
	}
	if(bb=0) cnt=1,w[1]=0;
	int ansb=0;
	ansb+=dpp(cnt,2,0);
	for(int i=1;i<w[cnt];i++)
		ansb+=dpp(cnt,0,i);
	ansb+=dpp(cnt,1,w[cnt]);
	ansb=max(ansb,1);
	ansa=max(ansa,1);
	printf("%d\n",ansa-ansb);
	return 0;
}