P2657 [SCOI2009]windy数 题解

博客园同步

原题链接

简要题意：

一个 相邻两个数字差的绝对值都 $\ge 2$ 且不含前导零 的数 被称为 “windy数”。问从 $a$ 到 $b$ 的 “windy数”的个数。

首先，我们考虑 $1$ ~ $n$ 的 “windy数” 的个数怎么求。

用 $f_{i,j}$ 表示有 $i$ 位，最高位为 $j$ 的方案数。

那么，从 $i-1\to i$ 只需要在 最高位前面拼上合法的一位。

即：

$$f_{i,j}=\sum _{k=0}^9{f}_{i-1,k}[\mathrm{abs}j,k\ge 2]$$

$k$ 即枚举合法的数字。

对于 $i=1$ 的情况，$f_{i,j}=1$. 则：

$$f_{i,j}=\{\begin{array}{l}{\sum }_{k=0}^9{f}_{i-1,k}[\mathrm{abs}j,k\ge 2],i\ne 1\\ 1,i=1\end{array}$$

如何求答案呢？

我们分为三类：

1. 位数比 $n$ 小的数。

2. 位数和 $n$ 一样，但最高位比 $n$ 小的数。

3. 位数和最高位都和 $n$ 一样，但比 $n$ 小的数。

第一部分，答案为 （$len$ 为 $x$ 的位数）

$$\sum _{i=1}^{len-1}\sum _{j=1}^9{f}_{i,j}$$

第二部分则枚举最高位即可。

第三部分有点复杂，只需要枚举位数和次高位即可。

时间复杂度：$O({\mathrm{siz}}^{2}n)$.

实际得分：$100pts$.

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;

inline int read(){char ch=getchar(); int f=1; while(ch<'0' || ch>'9') {if(ch=='-') f=-f; ch=getchar();}
	int x=0; while(ch>='0' && ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar(); return x*f;}

int n,m,a[15];
int f[15][15];
	
inline int calc(int x) { //计算 1 ~ x 的答案
	memset(a,0,sizeof(a));
	int len=0; while(x) a[++len]=x%10,x/=10;
	int ans=0;
	for(int i=1;i<len;i++) 
	for(int j=1;j<=9;j++) ans+=f[i][j]; //第一部分
	for(int i=1;i<a[len];i++) ans+=f[len][i]; //第二部分
	for(int i=len-1;i>=1;i--) {
		for(int j=0;j<a[i];j++)
			if(abs(j-a[i+1])>=2) ans+=f[i][j];
		if(abs(a[i+1]-a[i])<2) break;	//第三部分
	} return ans;
}

int main() {
	n=read(),m=read();
	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]; //求出 f
	printf("%d\n",calc(m+1)-calc(n));	
	return 0;
}