1026: [SCOI2009]windy数

1026: [SCOI2009]windy数

Time Limit: 1 Sec  Memory Limit: 162 MB
Submit: 8247  Solved: 3708
[Submit][Status][Discuss]

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

/**
 * @Date:   2017-09-15T21:47:32+08:00
 * @Last modified time: 2017-09-15T22:47:40+08:00
 */
/*
 题意：windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整
    数被称为windy数。 windy想知道，在A和B之间，包括A和B，总共有多少个windy
    数？

 思路：数位DP
*/

#include <bits/stdc++.h>
#include <stdio.h>

#define MAXNUM 20
#define MAXK 15
#define MAXF 3

using namespace std;

int n,m;
int L,R;
int num[MAXNUM];

int dp[MAXNUM][MAXK][MAXF][MAXF];//dp[i][j][k][l]表示k位，上一位是j时的方案数,是不是比规定的数小
//是不是第一个数

inline int dfs(int len,int last,bool flag,bool zero){//枚举到第几位，从高位到底位，上一位是什么，
    //是不是小于最大数,是不是第一个数字
    if(len==0)
        return 1;
    if(dp[len][last+2][flag][zero]!=0)
        return dp[len][last+2][flag][zero];
    int tol=(flag==true?num[len]:9);
    int res=0;
    for(int i=0;i<=tol;i++){
        if(abs(last-i)>=2||zero){
            res+=dfs(len-1,i,flag&i==tol,zero&&i==0);
        }
    }
    dp[len][last+2][flag][zero]=res;
    return res;
}

inline int cal(int x){
    memset(num,0,sizeof num);
    int pos=1;
    while(x){
        num[pos++]=x%10;
        x/=10;
    }
    return dfs(pos-1,-2,true,true);
}

inline void init(){
    L=0;
    R=0;
    memset(dp,0,sizeof dp);
}
int main(){//127322182
    // freopen("in.tx","r",stdin);
    scanf("%d%d",&n,&m);
    init();
    L=cal(n-1);
    R=cal(m);
    printf("%d",R-L);
    return 0;
}