[BZOJ1026] [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
感觉还是没有学会数位DP啊

#include <cstdio>
#include <cstring>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;
long long f[15][15];
void init()
{
    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 sa[100];
long long ask(long long x)
{
    sa[0]=0;
    while(x) sa[++sa[0]]=x%10,x/=10;
    long long ans=0;
    sa[sa[0]+1]=-1;
    for(int i=1;i<sa[0];i++)
        for(int j=1;j<=9;j++) ans+=f[i][j];
    for(int i=1;i<sa[sa[0]];i++) ans+=f[sa[0]][i];
    for(int i=sa[0]-1;i>0;i--)
    {
        for(int j=0;j<sa[i];j++)
            if(abs(j-sa[i+1])>=2) ans+=f[i][j];
        if(abs(sa[i]-sa[i+1])<2) break;
    }
    return ans;
}
int main()
{
    init();
    long long a,b;  
    scanf("%lld%lld",&a,&b);
    printf("%lld",ask(b+1)-ask(a));
    return 0;
}

发表评论

邮箱地址不会被公开。 必填项已用*标注