BZOJ 1026 [SCOI2009]windy数 - 数位DP

今天才学了数位DP，好开森，迫不及待地开始写题，写了一遍又一遍输出全是0，发现原来是函数没调用（233）

一道裸的数位dp题，利用类似前缀和差值计算。

其中warn含义为此位以及以前的位数均濒临上限，若下一位超f数组的限制需要break（转移也很好转移，当上一位有warn标记且f[len]==i）

而f数组则是给定数的每一位，即枚举的最大值。

由于位数不齐，所以需要利用start枚举开始的位数（递推则是此位start标记为假，当且仅当上一位标记为假且此位为0）.

差值为2的解决方法则是，若有start标记且abs<2则continue。

切记切记dp数组在计算时要归0。

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<algorithm>

using namespace std;

int n,l,r,ans,totlen;
int f[20];
int dp[20][2][15][2];

void init(int x)
{
while(x)f[++totlen]=x%10,x/=10;
}
int Search(int len,bool warn,int last,bool start)
{
int &res=dp[len][warn][last][start];

if(res!=-1)return res;
if(len==0)return res=1;

res=0;

for(int i=0;i<10;i++)
{
if(warn&&i>f[len])break;
bool nextstart=true,nextwarn;

if(!start&&i==0)nextstart=false;
if(abs(last-i)<2&&start)continue;

nextwarn=(warn&&(i==f[len]))?true:false;

res+=Search(len-1,nextwarn,i,nextstart);
}
return res;
}
int main()
{
scanf("%d%d",&l,&r);

memset(dp,-1,sizeof dp);
totlen=0;
init(r);
ans+=Search(totlen,1,0,0);

memset(dp,-1,sizeof dp);
totlen=0;
init(--l);
ans-=Search(totlen,1,0,0);

printf("%d",ans);
return 0;
}