bzoj 1026(luogu 2657) SCOI2009 windy数

题目

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，在A和B之间，包括A和B，总共有多少个windy数？

分析

一道显的数位DP，设f[i][j]表示当最高位为第i位且第i位为j时的windy数数目，则可列出状态转移方程f[i][j]=sum(f[i-1][k]) (abs(j-k)>1),注意要用long long，时间大约是50ms。

#include<bits/stdc++.h>
using namespace std;
long long a,b,f[20][20],xx[20];
void cl(){
    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];
}
long long solve(long long x){
    memset(xx,0,sizeof(xx));
    if(x==0) return 0;
    long long k=0,ans=0;
    while(x) xx[++k]=x%10,x/=10;
    for(int i=1;i<=k-1;i++) for(int j=1;j<=9;j++) ans+=f[i][j];
    for(int i=1;i<xx[k];i++) ans+=f[k][i];
    for(int i=k-1;i>=1;i--){
        for(int j=0;j<=xx[i]-1;j++) if(abs(j-xx[i+1])>=2) ans+=f[i][j];
        if(abs(xx[i+1]-xx[i])<2) break;
        if(i==1) ans += 1;
    }
    return ans;
}
int main(){
    memset(f,0,sizeof(f));
    cl();
    scanf("%lld%lld",&a,&b);
    printf("%lld",solve(b)-solve(a-1));
    return 0;
}