JZOJ 1039. 【SCOI2009】windy数

题目

Description

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

 

Input

两个整数，A B。

Output

一个整数，表示A~B中有多少个windy数。

 

Sample Input

1 10

Sample Output

9

 

Data Constraint

 

 

Hint

100%的数据，满足 1 <= A <= B <= 2000000000 。

 

分析

 

• 数位DP板子题

 

代码

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int e[10001],cnt;
long long f[12][12][2][2];
long long dp(int pos,int pre,bool zero,bool lim)
{
    if (pos<1) return 1;
    if (f[pos][pre][zero][lim]!=-1) return f[pos][pre][zero][lim];
    int end=lim?e[pos]:9;
    int ans=0;
    for (int i=0;i<=end;i++)
    {
        if (abs(i-pre)>=2||zero)
        {
            ans+=dp(pos-1,i,zero&&(i==0),lim&&(i==end));
        }
    }
    return f[pos][pre][zero][lim]=ans;
}
long long calc(int x)
{
    memset(f,-1,sizeof(f));
    memset(e,0,sizeof(e));
    cnt=0;
    while (x) e[++cnt]=x%10,x/=10;
    return dp(cnt,0,1,1);
}
int main ()
{
    int a,b;
    cin>>a>>b;
    cout<<calc(b)-calc(a-1);
}