bzoj 1026: [SCOI2009]windy数

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1026

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

解：数位dp，状态末尾的0是无限制，1是有限制

/*
 * Problem:  
 * Author:  SHJWUDP
 * Created Time:  2015/6/4 星期四 11:04:39
 * File Name: 233.cpp
 * State: 
 * Memo: 
 */
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long int64;

const int MaxA=10+7;

int f[MaxA][MaxA][2];    //f[][][1]:limit
int arr[MaxA], num;
int func(int x) {
    if(x<10) return x;
    num=0;
    do {
        arr[num++]=x%10;
        x/=10;
    } while(x);
    memset(f, 0, sizeof(f));
    for(int i=1; i<arr[num-1]; i++) f[num-1][i][0]=1;
    f[num-1][arr[num-1]][1]=1;
    for(int i=num-1; i>0; i--) {
        for(int j=0; j<=9; j++) {
            for(int k=0; k<=9; k++) {
                if(abs(j-k)>=2) {
                    f[i-1][k][0]+=f[i][j][0];
                    if(k<arr[i-1]) f[i-1][k][0]+=f[i][j][1];
                    else if(k==arr[i-1]) f[i-1][k][1]+=f[i][j][1];
                }
            }
            if(j!=0) f[i-1][j][0]++;    ///leading zeros
        }
    }
    int res=0;
    for(int i=0; i<=9; i++) {
        res+=f[0][i][0]+f[0][i][1];
    }
    return res;
}
int main() {
#ifndef ONLINE_JUDGE
    freopen("in", "r", stdin);
    //freopen("out", "w", stdout);
#endif
    int A, B;
    while(~scanf("%d%d", &A, &B)) {
        printf("%d
", func(B)-func(A-1));
    }
    return 0;
}