洛谷P2657 Loj10165 SCOI2009 windy数

题目描述

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，

在A和B之间，包括A和B，总共有多少个windy数？

输入输出格式

输入格式：

包含两个整数，A B。

输出格式：

一个整数

数位DP，记录每一位最后一位是什么，枚举时如果他们的差小于2就return。

但是注意！如果有前导零，也就是00014是一个合法的序列，但是一般会判断1和0差小于2，就算非法了。

我们需要记录前导零，如果当前是前导零，则默认当前位位-10000，也就是下一位随便取。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#define in(a) a=read()
#define REP(i,k,n)  for(int i=k;i<=n;i++)
using namespace std;
inline int read(){
    int x=,f=;
    char ch=getchar();
    for(;!isdigit(ch);ch=getchar())
        if(ch=='-')
            f=-;
    for(;isdigit(ch);ch=getchar())
        x=x*+ch-'';
    return x*f;
}
int a,b,digit[],dp[][],ind;
inline int dfs(int pos,int state,bool ze,bool flag){
    if(!pos)  return ;
    if(!ze && dp[pos][state] && !flag)  return dp[pos][state];
    int up,ans=;
    if(flag)  up=digit[pos];
    else  up=;
    REP(i,,up){
        if(abs(i-state)<)  continue;
        if(ze && !i)  ans+=dfs(pos-,-,,flag && i==digit[pos]);
        else  ans+=dfs(pos-,i,,flag && i==digit[pos]);
    }
    if(!flag && !ze)  dp[pos][state]=ans;
    return ans;
}
inline int solve(int x){
    ind=;
    while(x){
        digit[++ind]=x%;
        x/=;
    }
    memset(dp,,sizeof(dp));
    return dfs(ind,-,,);
}
int main(){
    in(a),in(b);
    printf("%d",solve(b)-solve(a-));
}