Description
windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道,在A和B之间,包括A和B,总共有多少个windy数?
Input
包含两个整数,A B。
Output
一个整数。
Sample Input
【输入样例一】
1 10
【输入样例二】
25 50
1 10
【输入样例二】
25 50
Sample Output
【输出样例一】
9
【输出样例二】
20
9
【输出样例二】
20
HINT
【数据规模和约定】
100%的数据,满足 1 <= A <= B <= 2000000000 。
数位dp,预处理长度为i,最高位为j的windy数,然后长度小于a的windy数,在处理长度相同的,在处理长度相同各个最高位不同的。
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <iostream>
#include <queue>
#include <stack>
using namespace std;
int f[12][10];
int dp[12];
void init()
{
memset(f,0,sizeof(f));
int i,j,k;
for(i=0; i<10; i++)
f[1][i]=1;
for(i=2; i<=10; i++)
{
for(j=0; j<10; j++)
{
for(k=0; k<10; k++)
{
if(abs(j-k)>=2)
f[i][j]+=f[i-1][k];
}
}
}
}
int solve(int a)
{
int i,j;
int len=0;
while(a)
{
dp[len++]=a%10;
a/=10;
}
int ans=0;
for(i=1; i<len; i++)
for(j=1; j<10; j++)
ans+=f[i][j];
for(i=1; i<dp[len-1]; i++)
ans+=f[len][i];
for(i=len-1; i>=1; i--)
{
for(j=0; j<dp[i-1]; j++)
if(abs(j-dp[i])>=2)
ans+=f[i][j];
if(abs(dp[i]-dp[i-1])<2)
break;
}
return ans;
}
int main()
{
init();
// int i,j;
// for(i=1; i<10; i++)
// {
// for(j=1; j<10; j++)
// {
// cout<<f[i][j]<<" ";
// }
// cout<<endl;
// }
int a,b;
while(scanf("%d%d",&a,&b)!=EOF)
{
int res=solve(b+1)-solve(a);
cout<<res<<endl;
}
return 0;
}