（数位dp模板）[洛谷 P2657] windy 数

解题思路

这是一道经典的数位dp模板题

根据题意，我么需要得知范围$[l,r]$内有多少个符合条件的数字，我们可以利用前缀和的想法，用$[1,r]$内的个数减去$[1,l]$内的个数

这就把问题转变为了如何快速的求出$[1,n]$范围内符合条件的个数

很显然暴力枚举一定会出现问题，所以要借助数位dp的构想，通过记忆化搜索，快速求出需要的答案

首先要把n这个数字逐个拆解,并初始化数组

int solve(int o)
{
	a.clear();
	do{
		a.push_back(o%10);
		o /= 10;
	}while (o);
	memset(dp,-1,sizeof dp);
	return dfs(a.size(),-2,1,1);
}

接下来介绍一下核心部分dfs

• pos : 记录一下此时检索的是第几位，从最高位开始
• last ：前一个数字是什么，由于第一位数字前面没有数字，又需要其正常进行，所以初始化为-2
• lead： 有无前导零，这里的实际意义是该位是不是目前的第一位，因为在遍历的过程中，很可能会出现，前面都是0的情况，true则代表前面全是0，意义就是现在检索的这一位是目前第一位
• limit：是否有限制，最大数是固定的n，假设n=12345，那么此时检索到第三位的时候，前面两位分别是1和0，此时第三位取0~9任意一个数字都是成立的，如果前两位是1和2，那么这一位范围就是0~3

int dfs(int pos, int last, int lead, int limit)
{
	int res = 0;
	//说明此时pos已经遍历完毕，到达此地就说明是个windy数
	if(!pos) return 1;
	//记忆化搜索，查过的就快速检索出来
	if(~dp[pos][last][lead][limit]) return dp[pos][last][lead][limit];
	//根据limit来判断是否有最大值
	int maxn = (limit ? a[pos-1] : 9);
	for(int i=0 ; i<=maxn ; i++){
		//如果不符合又不符合的i直接跳过
		if(abs(i - last) < 2) continue;
		
		//这里代表着前面都是0，此时的这位是目前这个数字的第一位
		if(lead && i == 0) res += dfs(pos-1 , -2 , 1 , limit && a[pos-1] == i);
		
		//这里说明一下a[pos-1] == i一定要结合i<=maxn来看
		else res += dfs(pos-1 , i , 0 , limit && a[pos-1] == i);
	}
	//记得存下，便于后续调用
	return dp[pos][last][lead][limit] = res;
}

总代码

#include<bits/stdc++.h>
using namespace std;
vector<int> a;
int dp[15][15][2][2];
int dfs(int pos, int last, int lead, int limit)
{
	int res = 0;
	if(!pos) return 1;
	if(~dp[pos][last][lead][limit]) return dp[pos][last][lead][limit];
	int maxn = (limit ? a[pos-1] : 9);
	for(int i=0 ; i<=maxn ; i++){
		if(abs(i - last) < 2) continue;
		if(lead && i == 0) res += dfs(pos-1 , -2 , 1 , limit && a[pos-1] == i);
		else res += dfs(pos-1 , i , 0 , limit && a[pos-1] == i);
	}
	return dp[pos][last][lead][limit] = res;
}
int solve(int o)
{
	a.clear();
	do{
		a.push_back(o%10);
		o /= 10;
	}while (o);
	memset(dp,-1,sizeof dp);
	return dfs(a.size(),-2,1,1);
}
int main()
{
	int a,b;
	cin >> a >> b;
	cout << solve(b) - solve(a-1) << endl;
	return 0;
}