windy数

Posted 2021-02-06 mch5201314

链接

[https://vjudge.net/contest/281140#problem/D]

题意

　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 。

分析

需要判断是否有前导零即可，具体看代码

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long
ll w[20],dp[20][10];
//dp[i][0,1]代表前一位是否为某个数的情况下，i位(0~i个9)满足条件的个数 
ll dfs(ll len,ll last,bool limit,bool zero){
    if(len==0) return 1;
    //注意 
    if(!zero&&!limit&&dp[len][last]) return dp[len][last];
    ll sum=0,maxn=(limit?w[len]:9); 
    for(ll i=0;i<=maxn;i++){
        if(abs(i-last)>=2&&!zero||zero) 
        sum+=dfs(len-1,i,limit&&i==maxn,zero&&i==0);
    }
    if(!limit&&!zero) return dp[len][last]=sum;
    return sum;
}
ll solve(ll x){
    ll j=0;
    memset(w,0,sizeof(w));
    while(x){
        w[++j]=x%10;
        x/=10;
    }
    return dfs(j,0,1,1);
}
int main(){
    ll l,r;
    while(~scanf("%lld%lld",&l,&r)){
        printf("%lld
",solve(r)-solve(l-1));
    }
    return 0;
}