数位DP模板。注意 s o l v e solve solve函数中, p o s pos pos要置为 0 0 0。这种写法要把 m e m s e t memset 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);
}
下面这种写法可以把 m e m s e t memset memset写在每组数据外面。因为当 l i m i t = = 0 limit==0 limit==0的时候才会记录在 d p dp dp数组里面。也就是说从那个位置开始后面都是没有限制的。不管是 l l l还是 r r r,到这里以后算出来都是一样的。如果 l i m i t = = 1 limit==1 limit==1的时候记下来了,就说明前面被限制,而对于不同的数,限制是不同的。当前位置选择的数也是有限的,后面统计出来的答案就不一样。所以记下来的 d p [ p o s ] [ s t a ] [ 1 ] dp[pos][sta][1] dp[pos][sta][1]是不通用的,而 d p [ p o s ] [ s t a ] [ 0 ] dp[pos][sta][0] 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));
}
}