[SCOI2009]windy数(数位dp)

 题目描述

 windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，

 在A和B之间，包括A和B，总共有多少个windy数？

 题解

b站传送门

代码

 

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

int c[22];
int dp[15][15];

int dfs(int len,int pre,bool flag,bool lead0)
{
    if(len==0) return 1;
    int max_num;
    if(!flag&&pre>0&&dp[len][pre]!=-1) return dp[len][pre];
    if(flag) max_num=c[len]; else max_num=9;
    int res=0;
    for(int i=0;i<=max_num;i++)
    {
        if(abs(i-pre)>=2||lead0) res+=dfs(len-1,i,flag&&(i==max_num),lead0&&(i==0));
    }    
    if(!flag) dp[len][pre]=res;
    return res;
}

int slove(int x)
{
    int len=0;
    while(x)
    {
        c[++len]=x%10;
        x=x/10;
    }
    return dfs(len,-2,1,1);
}

int main()
{
    memset(dp,-1,sizeof(dp));
    int a,b;
    scanf("%d%d",&a,&b);
    cout<<slove(b)-slove(a-1)<<endl;
    return 0;
}