[Luogu P2657] [BZOJ 1026] [SCOI2009]windy数

洛谷传送门

BZOJ传送门

题目描述

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为$2$的正整数被称为windy数。 windy想知道，

在$A$和$B$之间，包括$A$和$B$，总共有多少个windy数？

输入输出格式

输入格式：

包含两个整数，$AB$。

输出格式：

一个整数

输入输出样例

输入样例#1：

1 10

输出样例#1：

9

输入样例#2：

25 50

输出样例#2：

20

说明

$100\%$的数据，满足 $1\le A\le B\le 2000000000$。

解题分析

显然我们可以预处理出一共有$i$位， 最高位为$j$的$windy$数个数。 按照套路， 我们只需要得到$[1,lef-1]$的$windy$数个数和$[1,rig]$的$windy$数个数即可。现在我们只讨论$[1,k]$的$windy$数个数。

设$k$有$m$位， 最高位为$t$， 那么我们分以下情况考虑：

1. 位数$<m$的情况， 直接累加即可。
2. 位数$=m$， 但最高位$<t$的情况， 仍然直接累加。
3. 位数$=m$，最高位$=t$的情况， 考虑次高位为$x$， 直接累加次高位$0\sim x-1$的结果， 继续如此计算下去。 注意如果某两位原本的差就$<2$直接$break$， 因为我们考虑的是更高位取满的情况。

这样有一个小锅： 因为我们始终累加到当前考虑的最高位的数位$-1$（因为最高位可能取不满）， 所以实质上我们的$solve(n)$算出来的是$n-1$的结果， 预先$+1$即可。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <cmath>
#include <algorithm>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define ll long long
#define MX 12
template <class T> IN T abs(T a) {return a > 0 ? a : -a;}
ll dp[MX][10];
int buf[MX];
void init()
{
    for (R int i = 0; i <= 9; ++i) dp[1][i] = 1;
    for (R int i = 2; i <= 10; ++i)
    {
        for (R int j = 0; j <= 9; ++j)
        for (R int k = 0; k <= 9; ++k)
        if(abs(j - k) >= 2) dp[i][j] += dp[i - 1][k];
    }
}
int solve(R int lim)
{
    if(lim == 0) return 0;
    int ans = 0, len = 0;
    W (lim) buf[++len] = lim % 10, lim /= 10;
    for (R int i = len - 1; i; --i)
    for (R int j = 1; j <= 9; ++j) ans += dp[i][j];
    for (R int i = 1; i < buf[len]; ++i) ans += dp[len][i];
    for (R int i = len - 1; i; --i)
    {
        for (R int j = 0; j < buf[i]; ++j)
        if(abs(j - buf[i + 1]) >= 2) ans += dp[i][j];
        if(abs(buf[i + 1] - buf[i]) < 2) break;
    }
    return ans;
}
int main(void)
{
    int lef, rig;
    init();
    scanf("%d%d", &lef, &rig);
    printf("%d", solve(rig + 1) - solve(lef));
}