1026: [SCOI2009]windy数

1026: [SCOI2009]windy数

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 5882   Solved: 2625
[ 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 。

Source

[ Submit][ Status][ Discuss]
数位dp f[i][j]为第i位为j的windy数方案 转移显然 ans可以用区间减法统计 问题在如何去掉比最大值大的东西 先贴着原数走，看原数的前面最多几位符合windy数 设为k 那么f[k]以前的状态往后转移的时候，如果f[k+1]这一位上的数比原数第k+1位来得大 那么f[k+1]这边的这个状态要-1(因为有一个合法windy数太大啦)

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<bitset>
#include<algorithm>
#include<cstring>
#include<map>
#include<stack>
#include<set>
#include<cmath>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;

int A,B,len,num[15],f[15][10];

void Change(int now)
{
	len = 0;
	if (!now) return;
	for (;;) {
		num[++len] = now%10;
		now /= 10;
		if (!now) break;
	}
	for (int i = 1; i <= len/2; i++) 
		swap(num[i],num[len-i+1]);
}

int Solve()
{
	int tail = len;
	for (int i = 1; i < len; i++)
		if (abs(num[i] - num[i+1]) < 2) {
			tail = i;
			break;
		}
	memset(f,0,sizeof(f));
	for (int i = 1; i <= num[1]; i++) f[1][i] = 1;
	for (int i = 1; i < len; i++) {
		for (int j = 0; j <= 9; j++) {
			if (!f[i][j]) continue;
			for (int k = 0; k <= j-2; k++)
				f[i+1][k] += f[i][j];
			for (int k = j+2; k <= 9; k++)
				f[i+1][k] += f[i][j];
			if (j == num[i] && i <= tail) {
				for (int k = 0; k <= j-2; k++)
					if (k > num[i+1])
						--f[i+1][k];
				for (int k = j+2; k <= 9; k++)
					if (k > num[i+1])
						--f[i+1][k];
			}
		}
		for (int k = 1; k <= 9; k++)
			++f[i+1][k];
	}
	int ret = 0;
	for (int i = 0; i <= 9; i++) 
		ret += f[len][i];
	return ret;
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	cin >> A >> B;
	Change(A-1);
	int a1 = Solve();
	Change(B);
	int a2 = Solve();
	cout << a2 - a1;
	return 0;
}