uestc windy数

链接：http://acm.uestc.edu.cn/problem.php?pid=1307

数位dp，详细解释见代码注释

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
int dp[15][10];
//dp[i][j],区间[10^(i-1),10^i)内以j开头的windy数个数
void inidp()
{
	int i,j;
	memset(dp,0,sizeof(dp));
	for(i=0;i<=9;i++) dp[1][i]=1;
	for(i=2;i<=12;i++)
	{
		//在0前面可以放2到9...以此类推，写得丑
		for(j=2;j<=9;j++) dp[i][j]+=dp[i-1][0];
		for(j=3;j<=9;j++) dp[i][j]+=dp[i-1][1];
		dp[i][0]+=dp[i-1][2];
		for(j=4;j<=9;j++) dp[i][j]+=dp[i-1][2];
		for(j=0;j<=1;j++) dp[i][j]+=dp[i-1][3];
		for(j=5;j<=9;j++) dp[i][j]+=dp[i-1][3];
		for(j=0;j<=2;j++) dp[i][j]+=dp[i-1][4];
		for(j=6;j<=9;j++) dp[i][j]+=dp[i-1][4];
		for(j=0;j<=3;j++) dp[i][j]+=dp[i-1][5];
		for(j=7;j<=9;j++) dp[i][j]+=dp[i-1][5];
		for(j=0;j<=4;j++) dp[i][j]+=dp[i-1][6];
		for(j=8;j<=9;j++) dp[i][j]+=dp[i-1][6];
		for(j=0;j<=5;j++) dp[i][j]+=dp[i-1][7];
		dp[i][9]+=dp[i-1][7];
		for(j=0;j<=6;j++) dp[i][j]+=dp[i-1][8];
		for(j=0;j<=7;j++) dp[i][j]+=dp[i-1][9];
	}
}
int solve(int n)
{
	int len=0,bit[15],i,j;
	while(n){
		bit[++len]=n%10;
		n/=10;
	}
	bit[len+1]=0;
	int ans=0;
	//以n=422为例
	//[1,99]
	for(i=1;i<len;i++)
		for(int j=1;j<10;j++)
			ans+=dp[i][j];
	//1XX
	//2XX
	//3XX
	for(j=1;j<bit[len];j++)
		ans+=dp[len][j];
	//至此以遍历了[1,399]
	//还有[400,422]，注意到这个区间的数字最高位都是4，所以下面从第二高位开始
	//一共两次循环
	for(i=len-1;i;i--)
	{
		for(j=0;j<bit[i];j++)
			if(abs(j-bit[i+1])>=2)
				ans+=dp[i][j];
		//第一次：[400,419]
		//第二次:[420,422]
		if(abs(bit[i]-bit[i+1])<2)
			break;
	}
	//最后遍历的总区间为[1,422]
	return ans;
 }
int main()
{
	inidp();
	int a,b;
	while(scanf("%d%d",&a,&b)!=EOF)
	{
		printf("%d\n",solve(b+1)-solve(a));
	}
	return 0;
}