BZOJ-1026 (数位dp)

题目描述

题目链接

Description

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

Sample

编号	输入样例	输出样例
.#1	1 10	9
.#2	25 50	20
.#3	15745 51248621	5588988

HINT

100%的数据，满足 1 <= A <= B <= 2000000000 。

解题思路

嗯，据说，这是一道数位DP题，以前都没做过数位DP，也就是说，这是我做的第一道数位DP题……
首先，为了描述方便，不妨把 不含前导零且相邻两个数字之差至少为2的正整数称为Ha；

注意 Ha 有两个性质：

1： 若有一个Ha记作a1，那么Ha1在最高位左边再加一位数得到a2，只要它与Ha1的最高位的差不小于二，那么a2就也是一个Ha。

比如 a1=135，那么再加一个“千位”k成为a2，只需满足abs(k-1)>=2即可使a2为一个Ha。

2：若有一个数已经确定有相邻两位数之差小于二，那么它再怎么改(当然那两个相邻的数不能改)，它都不能变成一个Ha。

比如对于数276326，千位和万位之差小于二，那么保持这两位不变，怎么变都不能成为Ha了。

预处理出所有 i 位数中 Ha 的数量(运用性质1)

1. 定义一个数组dp[][]，dp[i][j]代表 i 位数中最高位为 j 的 Ha 的个数；
2. 初始化一下，把一位数 (即{dp[1][j]}，其中j=0~9) 都赋初始值 1 ，因为的确就只有一个；
3. 对每一个位数 i，枚举每一个最高位(即第i位) j，其中再枚举其次高位(即第i-1位) k，如果 abs(j-k)>=2，那么 dp[i][j]+=dp[i-1][k]。

做一个函数 haha(x) 代表 小于 x 的 Ha 的个数(运用性质2)

1. 把 x 每一位分别放入数组p[]内，p[i] 中 i 越大位数越高(1:个位 2:十位 3:百位…) ，顺便就求出了 x 的位数 n；
2. 最高位比 x 最高位小的 Ha 肯定比 x 小，sum+=Sigma[dp[n][i], {i, 1, p[n] - 1}]；
3. 之后处理最高位与 x 最高位相同的 Ha 的个数，把它们加到sum里 (具体见程序中的注释)；
4. return(sum)。

输出答案

注意是 haha(B+1)-haha(A)，因为haha(x)处理的是小于(不包括) x 的 Ha 的个数。

程序

#include <cstdio>
#include <algorithm>
#define LL long long
int dp[20][10];             //dp[i][j]: i 位数中最高位(最左边)为 j 符合的个数 
int A,B,p[11],n,sum;        //p[i]: 当前处理的数(i越大越高位) 

//小于 x 的 Ha 的个数 
int haha(int x){
    if (x==0) return 0;
    n=sum=0;
    while (x) p[++n]=x%10, x/=10;
    for (int i=1; i<n; i++) for (int j=1; j<=9; j++) sum+=dp[i][j];

    //最高位比 x 最高位小的 Ha 肯定满足 
    for (int i=1; i<p[n]; i++) sum+=dp[n][i];

    //最高位为 x 的 Ha 个数 (让第 (i+1) 及以上的位为它所能取的最大值，往低位处理) 
    for (int i=n-1; i>0; i--){                          //第 (i+1)~n 位都与 x 一样(即那几位取可取的最大值) 
        for (int j=0; j<p[i]; j++)                      //枚举第 i 位的数(当然要小于 p[i] ) 
            if (abs(j-p[i+1])>=2) sum+=dp[i][j];        //满足条件的话，就加一下答案 
        if (abs(p[i]-p[i+1])<2) break;                  //若 x 第 i~n 位有不满足条件的 "相邻数"，那么再往下都不能满足了
    }
    return sum;
}

int main(){
    scanf("%d%d",&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("%d",haha(B+1)-haha(A));                     //由于在 haha(x) 中，处理的一直是小于 x 的 Ha ，所以用 haha(B+1) - haha(A) 
}