bzoj 1026 Windy数 数位DP

bzoj 1026 Windy数

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

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 。

倒过来数位DP，f[i][j]为从右数第i位为j的Windy数个数

f[i][j]+=f[i-1][k](k<=j-2||k>=j+2)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#define MAXN 10

using namespace std;

int f[MAXN][10];
int len;

inline int abs(int x)
{
return x>0 ? x :-x;
}
inline void init()
{
for(int i=0; i<10; i++)
f[0][i]=1;
for(int i = 1; i < MAXN; i++)
{
for(int j=0; j<=9; j++)
{
f[i][j]=0;
for(int k=0; k<=j-2; k++)
f[i][j]+=f[i-1][k];
for(int k=j+2; k<=9; k++)
f[i][j]+=f[i-1][k];
}
}
}
inline bool isWindy(int n)
{

while(n>=10)
{
if(abs(n%10-n/10%10)<2)
return false;
n/=10;
}
return true;
}
inline int getsum(int n)
{
if(n==0)return 1;
int res=0;
if(isWindy(n))
res++;
int a[MAXN+1];
len=0;
while(n>0)
{
a[len++]=n%10;
n/=10;
}
int i;
for(i=len-1; i>=0&&(i>=len-2||abs(a[i+1]-a[i+2])>=2); i--)
{
if(i==len-1)
{
for(int j=1; j<a[i]; j++)
res+=f[i][j];
for (int j = i - 1; j >= 0; j--)
for (int k = 1; k <= 9; k++)
res += f[j][k];
res++;

}
else
{
for (int j = 0; j <= a[i + 1] - 2 && j < a[i]; j++)
res += f[i][j];
for (int j = a[i + 1] + 2; j < a[i]; j++)
res += f[i][j];
}
}
return res;
}

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