[SCOI2009]windy数 数位DP（记忆化搜索）

1026: [SCOI2009]windy数

Time Limit: 1 Sec  Memory Limit: 162 MB
Submit: 10876  Solved: 5122
[Submit][Status][Discuss]

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 。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<map>
#include<set>
#include<cmath>
#include<cctype>
#include<ctime>
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define MOD 998244353
typedef long long ll;
typedef unsigned long long ull;
const int mm=100000;

using namespace std;

int a[15],dp[15][15][2];

int dfs(int len,int pos,bool cnt,bool limit)
{
	if(len==-1) return 1;
	if(!limit&&dp[len][pos][cnt]!=-1) return dp[len][pos][cnt];
	int up=limit?a[len]:9;
	int ans=0;
	for(int i=0;i<=up;i++)
	{
		if(!cnt||abs(pos-i)>=2)
			ans+=dfs(len-1,i,i||cnt,i==a[len]&&limit);
	}
	if(!limit) dp[len][pos][cnt]=ans;
	return ans;
}

int fun(int x)
{
	int len=0;
	while(x)
	{
		a[len++]=x%10;
		x/=10;
	}
	return dfs(len-1,0,0,1);
}

int main()
{
	int n,m;
	while(cin>>n>>m)
	{
		memset(dp,-1,sizeof(dp));
		cout<<fun(m)-fun(n-1)<<endl;
	}
	return 0;
}