[luogu2657][windy数]

luogu2657

思路

数位dp，记录下上个位置的数，如果当前的数字与上个数字的差值小于2，就不再转移。还是要注意排除前导0。在记忆化的时候，全都是前导0的情况不能记忆化。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;

ll read() {
    ll x=0,f=1;char c=getchar();
    while(c<'0'||c>'9') {
        if(c=='-') f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9') {
        x=x*10+c-'0';
        c=getchar();
    }
    return x*f;
}
int a[100];
ll f[100][11];
int ab(int x) {
    return x<0?-x:x;
}
int dfs(int pos,int Rea,int limit,int lst) {
    if(!pos) return Rea;
    if(f[pos][lst]!=-1&&Rea&&!limit) return f[pos][lst];
    int up=limit?a[pos]:9;
    ll ans=0;
    for(int i=0;i<=up;++i) {
        if(ab(i-lst)<2&&Rea) continue;
        ans+=dfs(pos-1,Rea||i,limit&&(i==up),i);
    }
    if(!limit&&Rea) f[pos][lst]=ans;
    return ans;
}
int solve(ll x) {
    memset(f,-1,sizeof(f));
    memset(a,0,sizeof(a));
    while(x) {
        a[++a[0]]=x%10;
        x/=10;
    }
    return dfs(a[0],0,1,10);
}
int main() {
    ll x=read(),y=read();
//  cout<<solve(read());
    cout<<solve(y)-solve(x-1);
    return 0;
}

转载于:https://www.cnblogs.com/wxyww/p/9764404.html