题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=1026
题目大意:windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道,在A和B之间,包括A和B,总共有多少个windy数?
分析:数位DP,定义dp(i,j)为高一位为j的i位数中满足题意的数的个数。
实现代码如下:
#include <cstdio>
#include <cstring>
using namespace std;
int dp[25][25];
int dig[25],len;
int dfs(int pos,int pre,int z,int lim)
{
if(pos<=0) return 1;
if(z&&!lim&&dp[pos][pre]!=-1) return dp[pos][pre];
int num=lim?dig[pos]:9;
int ans=0;
if(z==0)
{
for(int i=0;i<=num;i++)
ans+=dfs(pos-1,i,i,lim&&(i==num));
}
else
{
for(int i=0;i<=num;i++)
{
if((i-pre)*(i-pre)<4) continue;
ans+=dfs(pos-1,i,1,lim&&(i==num));
}
}
if(z&&!lim) dp[pos][pre]=ans;
return ans;
}
int count(int x)
{
len=0;
while(x)
{
dig[++len]=x%10;
x/=10;
}
memset(dp,-1,sizeof(dp));
return dfs(len,0,0,1);
}
int main()
{
int l,r;
while(scanf("%d%d",&l,&r)!=-1)
{
printf("%d\n",count(r)-count(l-1));
}
return 0;
}