bzoj1026（数位dp）

来源：互联网发布：产品宣传动画制作软件编辑：程序博客网时间：2024/05/21 17:15

对拍了一下，刚开始wa了一次

这是要求处理没有前导0的，模版，其实就是原来的！limit的地方再加上！lead（就这里wa了一次）还有再加一个lead的参数就可以了，当lead为0时表示开始的情况，不要有限制，因为开始是随意的。

#include<cstdio>#include<cstring>#include<cmath>#include<cstdlib>#include<algorithm>using namespace std;typedef long long ll;ll f[20][12];int n,m,li[12];ll dfs(int i,bool lead,int state,bool limit){if (i==0) return 1;if (!limit && !lead && f[i][state]!=-1) return f[i][state];//这里就是说lead和limit的情况要特殊计算int up=limit ? li[i]:9;ll ans=0;for (int j=0;j<=up;j++)if (lead||abs(state-j)>=2){ans+=dfs(i-1,lead && j==0,j,limit && j==up);}if (!limit && !lead) f[i][state]=ans;return ans;}ll work(ll n){memset(f,-1,sizeof(f));memset(li,0,sizeof(li));int tot=0;while (n){li[++tot]=n%10;n=n/10;}return dfs(tot,true,0,true);//前导为0，所以上一个state也为0}int main(){scanf("%lld%lld",&n,&m);printf("%lld",work(m)-work(n-1));return 0;}

0 0