HYSBZ 1026 windy数(数位DP)

题目: 点这里

思路:
     第二次刷到这题，感觉比第一次写流畅许多,不过还是出了个傻逼bug找了好久，还是要找个时间集中刷一下。
      我数位DP采用的是记忆化搜索的形式(因为听说有结构体的数位dp)，这道数位DP感觉很板子，没啥难点，就是有几个小细节要注意，详情看代码.

AC_Code

#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
long long n,m;
int lt[15];
long long dp[15][15][5][5];//now ,bef ,limit , lead
int vis[15][15][5][5];

int getlens(long long x);
long long solve(long long x);
long long dfs(int now,int bef,bool limit,bool lead) ; // 表示当前的位数，前一位的数，是否是最高位限制，是否目前有前导零


int main() {
    cin>>n>>m;
    cout<< (solve(m)- solve(n-1)) <<endl;
    return 0;
}


int getlens(long long x) {
    if(x == 0) return 1;
    int len = 0;
    while(x!=0) {
        len++;
        lt[len] = x%10;
        x /= 10;
    }
    return len;
}

long long solve(long long x) {
    memset(dp,0,sizeof(dp));
    memset(vis,0,sizeof(vis));
    if(x == 0) return 1;
    int len = getlens(x);
    long long ans = 0;
    for(int i=0;i<=lt[len];i++) {
        ans += dfs(len-1,i,(i==lt[len]? 1:0),(i==0? 1:0));
    }
    //cout<<ans<<endl;
    return ans;    
}

long long dfs(int now,int bef,bool limit,bool lead) {
    // cout<<now<<' '<<bef<<' '<<limit<<' '<<lead<<endl;
    if(now == 0) return 1;
    if(vis[now][bef][limit][lead] == 1) return dp[now][bef][limit][lead];
    long long res = 0;
    int maxs = (limit? lt[now]:9);
    for(int i=(lead? 0:bef+2) ; i<=maxs;i++) { // 如果为 前导零，那就可以从 0 开始枚举
        res += dfs(now-1,i, (limit&&i==lt[now]? 1:0), (lead&& i==0? 1:0));
    }
    for(int i=min(bef-2,maxs);i>=0;i--) { // 可能 bef-2 > maxs
        res += dfs(now-1,i, (limit&&i==lt[now]? 1:0), (lead&& i==0? 1:0));
    }
    vis[now][bef][limit][lead] = 1;
    //cout<<now<<' '<<bef<<' '<<limit<<' '<<lead<<' '<<res<<endl;
    return dp[now][bef][limit][lead] = res;
}