【SCOI2009】【BZOJ1026】windy数

点击前往BZOJ原题
1026: [SCOI2009]windy数

Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 4078 Solved: 1827
[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的板题……用f[i][j]表示在第i位（从高到低）放j这个数windy数的个数。前导零的处理方法是某神犇教的……细节见代码。

ps:笔者近日做了许多COCI的题，里面的数位DP的题真的很不错推荐给大家，之前还对数位DP一窍不通，仅仅几天笔者回过头来看这种题一下就秒杀了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
int da[20];
LL f[20][20],ten[20];
void getnum(LL x,int &len){
    len=0;
    while(x){
        da[++len]=x%10;
        x/=10;      
    }
}
//DP用记忆化搜索实现
LL dp(int pos,int num,bool ex){
//ex表示前面位数的数是否到达上界
    if(f[pos][num]!=-1&&!ex)return f[pos][num];
//f数组记录的是没有到达上界的情况，若到达上界只能现场算了。尽管如此，笔者做的数位DP的题里这样写跑下来的时间都离时限较远，且这样编写也很方便。
    LL ret=0;
    if(pos==1)ret=1;
    else{
        int r=ex?da[pos-1]:9;
        for(int i=0;i<=r;i++){
            if(abs(num-i)>1){
                ret+=dp(pos-1,i,ex&(i==r));
            }
        }
    }
    if(!ex)f[pos][num]=ret;
    return ret;
}
LL num(LL x){
    if(!x)return 0;
    int len;
    getnum(x,len);
    LL ret=0;
    for(int i=1;i<=da[len];i++){
        ret+=dp(len,i,i==da[len]);
    }
    memset(f,-1,sizeof(f));
    /*前导零的处理方法楼主特别喜欢……设原数为k位数，则分别考虑k位数，k-1位数，k-2位数……1位数的情况，便轻松的忽略了前导零这种麻烦的东西。*/
    for(int i=len-1;i;i--){
        LL p=ten[i]-1;
        for(int j=1;j<=9;j++)ret+=dp(i,j,0);
    }
    return ret;
}
int main(){
    freopen("1026.in","r",stdin);
    LL a,b;
    cin>>a>>b;
    ten[0]=1;
    memset(f,-1,sizeof(f));
    for(int i=1;i<20;i++)ten[i]=ten[i-1]*10;
    //求区间[a,b]中windy数的个数即用[1,b]中windy数的个数减去[1,a-1]中windy数的个数。
    cout<<num(b)-num(a-1)<<endl;
    return 0;
}