[BZOJ1026][SCOI2009]windy数

1026: [SCOI2009]windy数

Time Limit: 1 Sec  Memory Limit: 162 MB
Submit: 7834  Solved: 3527
[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

HINT

【数据规模和约定】

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

裸裸的数位DP直接贴代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int f[15][15];
int find(int x)
{
    if(!x) return 0;
    int w=0;
    int t=x;
    int n[15];
    while(t)
    {
        n[++w]=t%10;
        t/=10;
    }
    int ans=0;
    for(int i=1;i<w;i++)
    {
        for(int j=1;j<=9;j++) ans+=f[i][j];
    }
    for(int i=w;i>=1;i--)
    {
        if(i==1)
        {
            for(int j=0;j<=n[i];j++)
                if(abs(n[i+1]-j)>=2) ans+=f[i][j];
        }
        else if(i==w)
            for(int j=1;j<n[i];j++) ans+=f[i][j];
        else
            for(int j=0;j<n[i];j++)
                if(abs(n[i+1]-j)>=2) ans+=f[i][j];
        if(w!=i) if(abs(n[i]-n[i+1])<=1) break;
    }
    return ans;
}
int main()
{
    for(int i=0;i<=9;i++) f[1][i]=1;
    for(int i=2;i<=10;i++)
        for(int j=0;j<=9;j++)
            for(int k=0;k<=9;k++)
                if(abs(j-k)>=2) f[i][j]+=f[i-1][k];
    int a,b;
    scanf("%d%d",&a,&b);
    printf("%d",find(b)-find(a-1));
}