BZOJ1026或洛谷2657 [SCOI2009]windy数

BZOJ原题链接

洛谷原题链接

简单的数位\(DP\)，套模板就好。

#include<cstdio>
#include<cstring>
using namespace std;
const int N = 15;
int f[N][N], a[N];
inline int re()
{
	int x = 0;
	char c = getchar();
	bool p = 0;
	for (; c < '0' || c > '9'; c = getchar())
		p |= c == '-';
	for (; c >= '0' && c <= '9'; c = getchar())
		x = x * 10 + c - '0';
	return p ? -x : x;
}
inline int jd(int x){ return x < 0 ? -x : x; }
int dfs(int pos, int pre, int lm, int zero)
{
	if (pos < 0)
		return 1;
	if (pre > 0 && !lm && !zero && f[pos][pre] > -1)
		return f[pos][pre];
	int i, k = lm ? a[pos] : 9, su = 0;
	for (i = 0; i <= k; i++)
		if (jd(i - pre) > 1)
			su += dfs(pos - 1, zero && !i ? -2 : i, lm && i == a[pos], zero && !i);
	if (pre > 0 && !lm && !zero)
		return f[pos][pre] = su;
	return su;
}
int calc(int x)
{
	int l = 0;
	while (x > 0)
	{
		a[l++] = x % 10;
		x /= 10;
	}
	return dfs(l - 1, -2, 1, 1);
}
int main()
{
	int x, y;
	x = re();
	y = re();
	memset(f, -1, sizeof(f));
	printf("%d", calc(y) - calc(x - 1));
	return 0;
}