bzoj1026 Windy数

传送门

数位DP模板。注意 $solve$函数中， $pos$要置为 $0$。这种写法要把 $memset$写在函数里面。

#include<bits/stdc++.h>
using namespace std;
int n,m,a[20],pos,f[20][10][2][2];
inline int dp(int pos,int pre,bool lead,bool limit,int ret=0){
    if(!pos) return 1;
    if(~f[pos][pre][lead][limit]) return f[pos][pre][lead][limit];
    int up=limit?a[pos]:9;
    for(int i=0;i<=up;++i){
        if(lead) ret+=dp(pos-1,i,i==0,limit&&(i==up));
        else if(abs(pre-i)>=2)ret+=dp(pos-1,i,0,limit&&(i==up));
    }return f[pos][pre][lead][limit]=ret;
}
inline int solve(int x){
    memset(f,-1,sizeof(f)),pos=0;
    while(x) a[++pos]=x%10,x/=10;
    return dp(pos,0,true,true);
} 
int main(){
    cin>>n>>m;
    cout<<solve(m)-solve(n-1);
}

下面这种写法可以把 $memset$写在每组数据外面。因为当 $limit==0$的时候才会记录在 $dp$数组里面。也就是说从那个位置开始后面都是没有限制的。不管是 $l$还是 $r$，到这里以后算出来都是一样的。如果 $limit==1$的时候记下来了，就说明前面被限制，而对于不同的数，限制是不同的。当前位置选择的数也是有限的，后面统计出来的答案就不一样。所以记下来的 $dp[pos][sta][1]$是不通用的，而 $dp[pos][sta][0]$是通用的。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int a[20],dp[20][2][2];
inline int dfs(int pos,int sta,int limit,int ret=0){
    if(!pos) return 1;
    if(~dp[pos][sta][limit]) return dp[pos][sta][limit];
    int up=limit?a[pos]:9;
    for(int i=0;i<=up;++i) if(i!=4){
        if(sta&&i==2) continue;
        ret+=dfs(pos-1,i==6,limit&&i==a[pos]);
    }
    if(!limit) dp[pos][sta][limit]=ret;
    return ret;
}
inline int solve(int x,int pos=0){
    while(x) a[++pos]=x%10,x/=10;
    return dfs(pos,0,1);
}
int main(){
    ll l,r;
    while(~scanf("%d%d",&l,&r)){
        memset(dp,-1,sizeof(dp));
        printf("%d\n",solve(r)-solve(l-1));
    }
}