BZOJ 1026: [SCOI2009]windy数 【数位dp】

Description

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，在A和B之间，包括A和B，总共有多少个windy数？

Input

包含两个整数，A B。

Output

一个整数。

Sample Input

【输入样例一】
1 10
【输入样例二】
25 50

Sample Output

【输出样例一】
9
【输出样例二】
20

HINT

 

【数据规模和约定】

100%的数据，满足 1 <= A <= B <= 2000000000 。

 

思路:数位dp水题,dp[pos][last][lead]表示考虑到第pos位，上一位数字是last，上一位数字是不是前导0,然后随便乱搞一下

 

#include<cstdio>
int abs(int n){if(n<0)return -n;else return n;}
using namespace std;
int dp[11][11][2];
int dfs(int pos,int last,int lead,int limit,int deci[])
{
    if(pos==0)return 1;
    //if(limit==0 && dp[pos][last][lead]!=-1)return dp[pos][last][lead];
    int temp = limit ? deci[pos] : 9;deci[0] = 0 ;
    for(int i = 0; i <= temp ; i++)
    {
        if(abs(i-last)>=2 || lead)
        {
            deci[0] += dfs(pos-1, i , i==0 && lead ,limit && i == temp ,deci);
        }
    }
    if(limit==0)dp[pos][last][lead] = deci[0];
    return deci[0];
}
int solve(int n)
{
    int deci[11]={0};
    while(n!=0)
    {
        deci[++deci[0]]=n%10;
        n/=10;
    }
    return dfs(deci[0],0,1,1,deci);
}
int main()
{
    int a,b;
    for(int i=0;i<=10;i++)
                for(int j=0;j<=10;j++)
                        for(int k=0;k<=1;k++)dp[i][j][k]=-1;
    scanf("%d%d",&a,&b);
    printf("%d",solve(b)-solve(a-1));
    return 0;
}