Bzoj P1026 [SCOI2009]windy数___数位dp

题目大意：

$windy$定义了一种$windy$数。不含前导零且相邻两个数字之差至少为$2$的正整数被称为$windy$数。
windy想知道，在$A$和$B$之间，包括$A$和$B$，总共有多少个$windy$数？
$1<=A<=B<=2000000000$

分析：

数位$dp$，
设$d{p}_{i,j}$表示选到了第$i$位，第$i$位选的数是$j$时的$windy$数总数。
不难发现$d{p}_{i,j}={\sum }_{k=0}^{9}d{p}_{i-1,k}(abs(j-k)\ge 2)$
然后我们可以发现，对于
询问的区间$[A,B]$内的$windy$数总数，
可以拆成区间$[1,B]$的$windy$数总数$-$区间$[1,A-1]$的$windy$总数
即$Su{m}_{[A,B]}=Su{m}_{[1,B]}-Su{m}_{[1,A-1]}$
然后我们现在就是要求对于一个数$x$，
在$[1,x]$内有多少个$windy$数，
我们将$x$十进制下的每一位拆出来，设位数为$len$，
那么我们分类讨论一下，
①对于最高的前$i$位，都填上$x$的前$i$位时的$windy$数个数
②位数小于$len$的数的$windy$数个数
③位数为$len$，最高位的数填上$<x$的最高位的能填的数时的$windy$个数
④它本身是否是个$windy$数

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <cstdlib>
#include <algorithm>

using namespace std;

int dp[20][10], Wxy[20], A, B, cnt, Total;

void Pre_Work()
{
	for (int i = 0; i <= 9; i++) dp[1][i] = 1;
	for (int i = 2; i <= 10; i++)
	    for (int j = 0; j <= 9; j++)
	        for (int k = 0; k <= 9; k++) 
			    if (abs(j - k) >= 2) dp[i][j] = dp[i][j] + dp[i - 1][k];
}

int Work(int x)
{
	cnt = 0, Total = 0;
	for (; x; x /= 10) Wxy[++cnt] = x % 10;
	for (int i = cnt - 1; i >= 0; i--)
	{
		if (i == 0) { ++Total; break; }
		for (int j = 0; j < Wxy[i]; j++) 
		    if (abs(j - Wxy[i + 1]) >= 2) Total += dp[i][j];
		if (abs(Wxy[i] - Wxy[i + 1]) < 2) break;
	}
	for (int i = 1; i < Wxy[cnt]; i++) Total += dp[cnt][i];
	for (int i = 1; i < cnt; i++)
	    for (int j = 1; j <= 9; j++) Total += dp[i][j];
    return Total;
}

int main()
{
	Pre_Work();
	scanf("%d %d", &A, &B);
	printf("%d\n", Work(B) - Work(A - 1));
	return 0;
}