数位dp--[SCOI2009]windy数

bzoj1026

数位dp常规题

首先数位dp，常见套路就是前缀和思想，用ans(b)-ans(a-1)表示答案

于是我们只要考虑从0到x的答案

---

数位dp通常会给你一个很大的数字，
我们要先把它用数组存下来，再从高位到低位考虑每一位
从高位到低位考虑的过程中就有高位对低位的限制，
也就是不能超过x，如果高位的数字与x前几位相同，那么这一位就不能填超过x这一位的数字
除了高位对低位的限制，通常还有相邻两位或几位或者前几位填的某些数对后面的限制
比如这道题中就是相邻两数差的绝对值必须>=2
那么我们只要在一位一位地考虑过程中把这个限制考虑进去就好了
而这个过程通常需要以递归的方式来实现，前面一般要预处理出每一位往后无高位对低位限制的方案数

那么整个过程就明朗了

---

这道题中我们设dp[i][j]表示一共i位，最高位为j的范围内的windy数
这个需要预处理出来
solve(x)表示0-x范围内有多少windy数
首先把x用数组存下并记录长度就不用说了
然后将1位到len-1位的答案加起来，这个是一定不会超过限制的
考虑共有len位的时候，最高位填1~num[len]-1的数也都不会超过限制，并且后面可以随便填，再考虑最高位是num[len]的时候，那么第len-1位只能填不超过num[len-1]的数字，这是一个类似于递归的过程，也是为什么数位dp通常用递归来实现的原因之一，但这道题里只需要循环一遍即可

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn 15
#define LL long long
using namespace std;
LL A,B,dp[maxn][10],num[maxn];

LL solve(LL x){
    if(x==0) return 0;
    LL ans=0; int len=0;
    memset(num,0,sizeof num);
    while(x){
        num[++len]=x%10;
        x/=10;
    }
    for(int i=1;i<=len-1;i++)
        for(int j=1;j<=9;j++)
            ans+=dp[i][j];
    for(int i=1;i<num[len];i++)
        ans+=dp[len][i];
    for(int i=len-1;i;i--){
        for(int j=0;j<num[i];j++)
            if(abs(j-num[i+1])>=2)
                ans+=dp[i][j];
        if(abs(num[i+1]-num[i])<2) break;
        if(i==1) ans++;
    }
    return ans;
}

int main(){
    scanf("%lld%lld",&A,&B);
    for(int i=0;i<=9;i++) dp[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) dp[i][j]+=dp[i-1][k];//预处理 
    printf("%lld\n",solve(B)-solve(A-1));
    return 0;
}