BZOJ 1026 windy数 （数位DP）

题目链接：BZOJ 1026

我开始是用记忆化搜索写的，然后前导零处理挂了。然后改用数位DP的一般形式写的。写数位DP的一般过程可以总结为先预处理符合条件的没有限制的数，然后将小于限制的直接相加，然后再处理边界上的数（边界处理各种蛋疼）。数位DP的一个重要的思想就是逐位计算。

两篇比较好的国家集训队论文：浅谈数位统计类问题            数位计数问题解法研究

然后是我的代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
using namespace std;

int base[20],f[20][20],dp[20][20];

inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}

void init(){
base[1]=1;//每一位的位权
for(int i=2;i<=10;i++)base[i]=base[i-1]*10;
for(int i=0;i<=9;i++)f[1][i]=1;//预处理每一位没有限制时符合条件的数的个数
for(int i=2;i<=10;i++)
for(int j=0;j<=9;j++)
for(int k=0;k<=9;k++)
if(abs((double)j-k)>=2)f[i][j]+=f[i-1][k];
}

int find(int x){
if(!x)return 0;
int t=0,w=10;
while(base[w]>x)w--;//找到x是几位数
for(int i=1;i<w;i++)//小于该位的没有前导零的方案数直接相加
for(int j=1;j<=9;j++)
t+=f[i][j];
//////////////////////////////////////////////////////////////
//考虑是w位数的情况
int limit=x/base[w];//limit为该位的限制
for(int i=1;i<limit;i++)t+=f[w][i];
x%=base[w];
int pre=limit;
for(int i=w-1;i>=1;i--){
limit=x/base[i];
if(i!=1){
for(int j=0;j<limit;j++)
if(abs((double)pre-j)>=2)t+=f[i][j];
}
else{
for(int j=0;j<=limit;j++)
if(abs((double)pre-j)>=2)t+=f[i][j];
}
if(abs((double)pre-limit)<2)break;//不合限制直接退出循环防止多算
x%=base[i];
pre=limit;
}
return t;
}

int main(){
init();
int A=read(),B=read();
printf("%d\n",find(B)-find(A-1));
return 0;
}

记忆化搜索写的先挖个坑之后再填吧。。。