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 。

 

题解：

设f[i][j]表示i位数第一位是j的windy数有多少，注意一下前导0

code：

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
char ch;
bool ok;
void read(int &x){
    for (ok=0,ch=getchar();!isdigit(ch);ch=getchar()) if (ch=='-') ok=1;
    for (x=0;isdigit(ch);x=x*10+ch-'0',ch=getchar());
    if (ok) x=-x;
}
const int maxl=10;
const int maxnum=10;
int l,r,f[maxl][maxnum],a[maxl];
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 calc(int n){
    int t=n,len=0,ans=0;
    memset(a,0,sizeof(a));
    while (t) a[++len]=t%10,t/=10;
    for (int i=len;i;i--){
        for (int j=0;j<a[i];j++) if (i+1>len||abs(j-a[i+1])>=2){
            if (i==len&&!j){
                for (int p=len-1;p;p--) for (int q=1;q<=9;q++) ans+=f[p][q];
            }
            else ans+=f[i][j];
        }
        if (i+1<=len&&abs(a[i+1]-a[i])<2) break;
    }
    return ans;
}
int main(){
    init();
    read(l),read(r);
    printf("%d
",calc(r+1)-calc(l));
    return 0;
}