Windy 数 LibreOJ - 10165 每一位比前一位大2，数位DP，寒假集训

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

Windy 想知道，在 A 和 B 之间，包括 A 和 B，总共有多少个 Windy 数？

输入格式
一行两个数，分别为 A,B。

输出格式
输出一个整数，表示答案。

样例 1
Input Output
1 10
9
样例 2
Input Output
25 50
20
数据范围与提示
20% 的数据，满足 1≤A≤B≤106；
100% 的数据，满足 1≤A≤B≤2×109。

直接看代码吧，

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <map>
#include <stack>
#include <set>
#include <queue>
#include <vector>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdio>
#define ls (p<<1)
#define rs (p<<1|1)
#define ll long long
using namespace std;
const int maxn = 2e6+5;
const int INF = 0x3f3f3f3f;
ll dp[50][15];//dp[pos][pre]记录了遍历第pos为位时，前一位为pre时的状态数
ll a[30],b[30];
// void init(){
//     dp[0][0]=1;
//     dp[0][1]=dp[0][2]=0;
//     for(int i=1;i<25;i++){
//         dp[i][0]=10*dp[i-1][0]-dp[i-1][1];//长度为i不含有49的个数
//         dp[i][1]=dp[i-1][0];//最高位为9但不含49的个数
//         dp[i][2]=10*dp[i-1][2]+dp[i-1][1];//含有49的个数
//     }
// }
ll dfs(ll pos,ll pre,bool limit){//pos表示当前遍历的是第几位，pre表示前一位是几,limit为限制标志 
    if(pos==-1) return 1;//找到一个没有49的数
    if(!limit&&dp[pos][pre]!=-1)
        return dp[pos][pre];
    ll up;
    if(limit)
        up=a[pos];//确定当前数的最大值
    else up=9;//无限制
    ll ans=0;
    for(int i=0;i<=up;i++){
        if(abs(pre-i)<2) continue;
        if(pre==11&&i==0)
            ans+=dfs(pos-1,11,limit&&i==a[pos]);//向下搜素并且
        else
            ans+=dfs(pos-1,i,limit&&i==a[pos]);
    }
    if(!limit)  
        dp[pos][pre]=ans;//无限制更新
    return ans;
}

ll init(ll n)
{
    ll len = 0;
    while(n) {
        a[len++] = n % 10;
        n /= 10;
    }
    return dfs(len-1,11,1);
}

void solve(){
    ll x,y;
    memset(dp,-1,sizeof(dp));
    while(scanf("%lld %lld",&x,&y)!=EOF){
        if(x==0&&y==0) break;
        cout<<init(y)-init(x-1)<<endl;
    }
}
int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
    return 0;
}