题目描述:
Description
windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道,
在A和B之间,包括A和B,总共有多少个windy数?
Input
包含两个整数,A B。
Output
一个整数
Sample Input
【输入样例一】
1 10
【输入样例二】
25 50
Sample Output
【输出样例一】
9
【输出样例二】
20
HINT
【数据规模和约定】
100%的数据,满足 1 <= A <= B <= 2000000000 。
题解:
此题显然是数位DP,f[i][j]表示前i位最高位是j的答案数。写一个记忆化搜索来做这个DP就行啦。。。
代码如下:
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
const int maxn=25,maxm=15;
int num[maxn],f[maxn][maxm];
int abs_(int x) {if (x<0) x=-x; return x;}
int dfs(int len,int last,bool pd){
if (len<=0) return 1;
if (!pd&&f[len][last]!=-1&&last>=0) return f[len][last];
int p,ret=0,maxx=(pd?num[len]:9);
for (int i=0;i<=maxx;i++) {
if (abs_(i-last)<2) continue;
if (i==0&&last==-10) p=last; else p=i;
ret+=dfs(len-1,p,pd&&(i==maxx));
}
if (last>=0&&!pd) f[len][last]=ret;
return ret;
}
int get_dp(int x){
int len=0;
while (x) num[++len]=x%10,x=x/10;
memset(f,255,sizeof(f));
return dfs(len,-10,1);
}
int main(){
int x,y; scanf("%d %d",&x,&y);
printf("%d\n",get_dp(y)-get_dp(x-1));
}