UESTC 250 windy数

人生第一个数位dp成就达成!


一直在看数位dp,一直没看懂,今天对着别人的板子硬敲了一发,突然感觉自己贯通了一些什么东西

光看是不行的,还是需要实际的在电脑上实现一下才有感觉


感觉数位dp对于普通的dp来说,状态的标记变量有点多,其它的也还好(也有可能是只做了一个简单的题,所以对数位dp这样的姿势的难度等级underfitting了


这一题的题解到处都是,我就不献丑了


#include<bits/stdc++.h>
using namespace std;

#define LL long long

LL dp[15][10][2];
int dig[15];

LL dfs(int pos,int pre,bool limit,bool leadz){
    if(pos < 0)
        return 1;
    if(!limit && dp[pos][pre][leadz]!=-1)
        return dp[pos][pre][leadz];
    LL ret = 0;
    int bound = limit ?dig[pos]:9;
    for(int i=0;i<=bound;i++){
         if(leadz || abs(i-pre)>=2)
            ret += dfs(pos-1,i,limit && (i==bound),leadz && (i==0));
    }
    if(!limit)
        dp[pos][pre][leadz] = ret;
    return ret;
}

LL caler(int n){
    int len = 0;
    memset(dp,-1,sizeof(dp));
    while(n){
        dig[len++] = n % 10;
        n /= 10;
    }
    return dfs(len-1,0,true,1);
}

int main(){
    LL a,b;
    while(cin>>a>>b){
        cout<<caler(b)-caler(a-1)<<endl;
    }
    return 0;
}


你可能感兴趣的:(UESTC 250 windy数)