UESTC 250 —— windy数 [入门题]

http://acm.uestc.edu.cn/#/problem/show/250

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>

using namespace std;
int a, b;
int dp[11][11];
int digit[11];

void init()
{
	memset(dp, 0, sizeof dp);
	for(int i = 0;i<10;i++)
		dp[1][i] = 1;
	for(int i = 2;i<=10;i++)
	{
		for(int j = 0;j<10;j++)
		{
			for(int k = 0;k<10;k++)
			{
				if(abs(j-k) >= 2)
				dp[i][j] += dp[i-1][k];
			}
		}
	}
}

int cal(int n)
{
	int len = 0;
	int ans = 0;
	while(n)
	{
		digit[++len] = n % 10;
		n /= 10;
	}
	digit[len+1] = 11;
	for(int i = 1;i<len;i++)
	{
		for(int j = 1;j<=9;j++)
		ans += dp[i][j];
	}
	for(int i = 1;i<digit[len];i++)
		ans += dp[len][i];
	int i;
	for(i = len-1;i>=1;i--)
	{
		for(int j = 0;j<digit[i];j++)
		{
			if(abs(j - digit[i+1]) >= 2)
			ans += dp[i][j];
		}
		if(abs(digit[i] - digit[i+1]) < 2)
		break;
	}
	if(i == 0 && abs(digit[1]-digit[2]) >= 2)	ans++;
	return ans;
}

int main()
{
	init();
	while(cin>>a>>b)
	{
		int n = cal(b);
		int m = cal(a-1);
		cout<<n-m<<endl;
	}
	return 0;
}