BZOJ1026(SCOI2009)[windy数]--数位DP

【链接】
bzoj1026

【解题报告】
简单的数位DP。

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=12;
int ans,n,a[maxn],b[maxn],f[maxn][10][2];
int Dfs(int x,int num,bool flg,bool pd,int c[])
{
    if (x>c[0]) return 1;
    if (!pd&&f[x][num][flg]) return f[x][num][flg];
    int MAX=9; if (pd&&c[x]<MAX) MAX=c[x]; f[x][num][flg]=0;
    for (int i=0; i<=MAX; i++)
     if ((!flg)||num-i>1||num-i<-1) f[x][num][flg]+=Dfs(x+1,i,flg|(i),pd&(i==MAX),c);
    return f[x][num][flg];
}
void Get(int x) {if (!x) return; Get(x/10); a[++a[0]]=x%10;}
int main()
{
    freopen("1026.in","r",stdin);
    freopen("1026.out","w",stdout);
    scanf("%d",&n); n--;
    if (!n) a[0]=1,a[1]=0; else Get(n);
    char ch=getchar(); b[0]=0;
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') b[++b[0]]=ch-48,ch=getchar();
    memset(f,0,sizeof(f));
    ans=Dfs(1,0,0,1,a);
    memset(f,0,sizeof(f));
    printf("%d\n",Dfs(1,0,0,1,b)-ans);
    return 0;
}