BZOJ 1026 1026: [SCOI2009]windy数

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，预处理长度为i，最高位为j的windy数，然后长度小于a的windy数，在处理长度相同的，在处理长度相同各个最高位不同的。

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <iostream>
#include <queue>
#include <stack>
using namespace std;
int f[12][10];
int dp[12];
void init()
{
    memset(f,0,sizeof(f));
    int i,j,k;
    for(i=0; i<10; i++)
        f[1][i]=1;
    for(i=2; i<=10; i++)
    {
        for(j=0; j<10; j++)
        {
            for(k=0; k<10; k++)
            {
                if(abs(j-k)>=2)
                    f[i][j]+=f[i-1][k];
            }
        }
    }
}
int solve(int a)
{
    int i,j;
    int len=0;
    while(a)
    {
        dp[len++]=a%10;
        a/=10;
    }
    int ans=0;
    for(i=1; i<len; i++)
        for(j=1; j<10; j++)
            ans+=f[i][j];
    for(i=1; i<dp[len-1]; i++)
        ans+=f[len][i];
    for(i=len-1; i>=1; i--)
    {
        for(j=0; j<dp[i-1]; j++)
            if(abs(j-dp[i])>=2)
                ans+=f[i][j];
        if(abs(dp[i]-dp[i-1])<2)
            break;
    }
    return ans;
}
int main()
{
    init();
//    int i,j;
//    for(i=1; i<10; i++)
//    {
//        for(j=1; j<10; j++)
//        {
//            cout<<f[i][j]<<" ";
//        }
//        cout<<endl;
//    }
    int a,b;
    while(scanf("%d%d",&a,&b)!=EOF)
    {
        int  res=solve(b+1)-solve(a);
        cout<<res<<endl;
    }
    return 0;
}