P2657 [SCOI2009]windy数

题目描述

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，

在A和B之间，包括A和B，总共有多少个windy数？

输入输出格式

输入格式：

包含两个整数，A B。

输出格式：

一个整数

#include<bits/stdc++.h>
using namespace std;
int a,b,l[32];
long long int f[50][10];
#define re(j) for(int j=0;j<=9;j++)
int work(int v)
{
    int len=0,ans=0;
    while(v){l[++len]=v%10;v=v/10;}
    for(int i=1;i<=len-1;i++)
    for(int j=1;j<=9;j++)ans+=f[i][j];
    for(int i=1;i<l[len];i++)
    ans+=f[len][i];
    for(int i=len-1;i>=1;i--)
    {
        for(int j=0;j<=l[i]-1;j++)if(abs(j-l[i+1])>=2)ans+=f[i][j];
        if(abs(l[i+1]-l[i])<2)break;
    }
    return ans;
}
int main()
{
    cin>>a>>b;
    for(int i=0;i<=9;i++)f[1][i]=1;
    for(int i=2;i<=10;i++)
    re(j)re(k)
    if(abs(j-k)>=2)f[i][j]+=f[i-1][k];
    cout<<(work(b+1)-work(a));
}