【BZOJ】1026: [SCOI2009]windy数（数位dp）

http://www.lydsy.com/JudgeOnline/problem.php?id=1026

我果然很弱啊。。。

考虑数位dp。枚举每一位，然后限制下一位即可。

一定要注意啊！在dfs的时候line这个要&&啊。。。。要不然wa了两发。。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
#define rep(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define read(a) a=getint()
#define print(a) printf("%d", a)
#define dbg(x) cout << (#x) << " = " << (x) << endl
#define printarr2(a, b, c) for1(_, 1, b) { for1(__, 1, c) cout << a[_][__]; cout << endl; }
#define printarr1(a, b) for1(_, 1, b) cout << a[_] << '\t'; cout << endl
inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; }
inline const int max(const int &a, const int &b) { return a>b?a:b; }
inline const int min(const int &a, const int &b) { return a<b?a:b; }

const int N=25;
int a[N], d[N][10], len;
int dfs(int x, int num, int front, int line) {
	if(!x) {
		if(front) return 0; //注意前导0不要，其实要不要都可以a。。。因为重复的会被剪掉。。
		return 1;
	}
	if(!front && !line && d[x][num]!=-1) return d[x][num];
	int last=9, tot=0; if(line) last=a[x];
	for1(i, 0, last) {
		if(front) {
			if(i==0) tot+=dfs(x-1, i, 1, line && i==last); //这个line一定要并上啊QAQ
			else tot+=dfs(x-1, i, 0, line && i==last);
		}
		else if(abs(i-num)>=2) tot+=dfs(x-1, i, 0, line && i==last);
	}
	if(!front && !line) d[x][num]=tot;
	return tot;
}
int getans(int x) {
	len=0;
	while(x) a[++len]=x%10, x/=10;
	return dfs(len, 0, 1, 1);
}
int main() {
	CC(d, -1);
	int x=getint(), y=getint();
	print(getans(y)-getans(x-1));
	return 0;
}

---

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 。

Source