BZOJ_P1026 [SCOI2009]windy数(动态规划)

BZOJ传送门
Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 4944 Solved: 2232
[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 。

Source

水题

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
int f[15][10];int a,b,l1,l2,ans;char s1[15],s2[15];
int main(){
    for(int i=0;i<10;i++) f[1][i]=1;
    for(int i=2;i<=10;i++)
        for(int j=0;j<10;j++){
            for(int k=0;k<10;k++)
                if(fabs(j-k)>1)
                    f[i][j]+=f[i-1][k];
        }

    scanf("%d%d",&a,&b);b++;
    sprintf(s1+1,"%d",a);sprintf(s2+1,"%d",b);
    l1=strlen(s1+1),l2=strlen(s2+1);
    reverse(s1+1,s1+l1+1),reverse(s2+1,s2+l2+1);

//  cout<<a<<" "<<b<<endl;
//  cout<<s1+1<<" "<<s2+1<<endl;
//  cout<<l1<<" "<<l2<<endl;


    for(int i=1;i<l2;i++)
        for(int j=1;j<10;j++)
            ans+=f[i][j];

//  cout<<ans<<endl;

    for(int i=1;i<s2[l2]-'0';i++) ans+=f[l2][i];

//  cout<<ans<<endl;

    for(int i=l2-1;i;--i){
        for(int j=0;j<s2[i]-'0';++j)
            if(fabs(s2[i+1]-'0'-j)>1)
                ans+=f[i][j];
        if(fabs(s2[i]-s2[i+1])<2) break;
    }

//  cout<<ans<<endl;

    for(int i=1;i<l1;i++)
        for(int j=1;j<10;j++)
            ans-=f[i][j];

//  cout<<ans<<endl;

    for(int i=1;i<s1[l1]-'0';i++) ans-=f[l1][i];

//  cout<<ans<<endl;

    for(int i=l1-1;i;--i){
        for(int j=0;j<s1[i]-'0';++j)
            if(fabs(s1[i+1]-'0'-j)>1)
                ans-=f[i][j];
        if(fabs(s1[i]-s1[i+1])<2) break;
    }

//  cout<<ans<<endl;

    printf("%d\n",ans);
    return 0;
}