BZOJ 1026([SCOI2009]windy数-数位DPlevel up)

1026: [SCOI2009]windy数

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 1280   Solved: 558
[ 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

【数据规模和约定】
20%的数据，满足 1 <= A <= B <= 1000000 。
100%的数据，满足 1 <= A <= B <= 2000000000 。

HINT
Source

[ Submit][ Status][ Discuss] 

数位DP

从前向后更新。

要分为1.当前位降；2.之前位降；3.前导0（因为要满足限制，所以索性直接递归999，99，9）

4.要包含当前数（要不然9999+1=10000，不能降位查）

一般情况下，数位DP是考虑不了=x的情况，而在某些时候一定得等（比如这题），大家各种特判吧……

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<functional>
#include<iostream>
#include<cmath>
#include<cctype>
#include<ctime>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define ForkD(i,k,n) for(int i=n;i>=k;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define MAXA (2000000000)
#define MAXN (1000+10)
long long A,B;
long long f[MAXN][10];
int a[MAXN];
long long calc(long long x)
{
    if (x<10) return x;
    Rep(i,MAXN) Rep(j,19) f[i][j]=0;
    int len=0;
    while (x) a[++len]=x%10,x/=10;
    For(i,a[len]-1) f[len][i]=1;
    bool flag=1;
    ForD(i,len)
    {
        if (i<len&&flag) Rep(j,a[i]) if (abs(a[i+1]-j)>=2) f[i][j]++;
        if (i<len&&abs(a[i]-a[i+1])<2) flag=0;
        Rep(j,10)
        {
            Rep(k,10)
                if (abs(j-k)>=2) f[i-1][k]+=f[i][j];
        }
    }
    long long ans=0;
    Rep(i,10) ans+=f[1][i];
    /* x本身*/
    flag=1;
    For(i,len-1) if (abs(a[i]-a[i+1])<2) {flag=0;break;}
    ans+=flag;


    if (len>1)
    {
        long long tmp=9;
        For(i,len-2) tmp=tmp*10+9;
        ans+=calc(tmp);
    }


    return ans;
}
int main()
{
//	freopen("bzoj1026.in","r",stdin);
    while (scanf("%d%d",&A,&B)==2)
    {
        cout<<calc(B)-calc(A-1)<<endl;

    }
	return 0;
}