1587 例题3 [SCOI2009] Windy 数(Bzoj1026 LOJ LUOGU2657 提高+/省选-) 需考虑前导0的数位DP

mrcrack

于 2022-04-21 17:07:11 发布

阅读量212

点赞数

分类专栏：信息学奥赛一本通提高篇文章标签：动态规划

本文链接：https://blog.csdn.net/mrcrack/article/details/123474386

版权

信息学奥赛一本通提高篇专栏收录该内容

41 篇文章 11 订阅

订阅专栏

需考虑前导0的数位DP

以下内容，可以结合后续的AC代码进行阅读。

dp[pos][pre]代表的是什么意思?

pos表示当前遍历的是第几位，pre表示前一位是几

dp[pos][pre]就是记录了遍历第pos位时，前一位为pre时的状态数

举例子
假设数5762，那么数位有4位
数位数组是这样存储的2 6 7 5
数组从0位开始，
所以是 0位到 3位
那么当pos为2的时候，前一位(即第3位)有0 - 5这些情况
那么dp[2][0-5]分别存储了dp[2][0],dp[2][1]。。。。等等这些情况

记搜过程
从起点向下搜索，到最底层得到方案数，一层一层向上返回答案并累加，最后从搜索起点得到最终答案。

对于[l,r]区间问题，我们一般把他转化为两次数位dp,即找[0,r]和[0,l−1]两段，再将结果相减就得到了我们需要的[l,r]

状态设计
如果理解了上述过程，我们需要考虑的就是怎样判断现在在哪一层，怎样判断当前的状态——这就需要我们传进一些参量。

dfs函数需要哪些参量？

首先是数位dp基本的量数字位数pos，最高位限制limit

由于数位dp解决的大多是数字组成问题，所以经常要比较当前位和前一位或前几位的关系（根据题意而定），所以一般在dfs()中也要记录前一位或前几位数pre方便比较。

除此之外还可以传进更多参量以区分状态，视题意而定。

最高位标记limit
我们知道在搜索的数位搜索范围可能发生变化；

举个例子：我们在搜索[0,567]的数时，显然最高位搜索范围是0~5，而后面的位数的取值范围会根据上一位发生变化：

当最高位是1~4时，第二位取值为[0,9];
当最高位是5时，第二位取值为[0,6]（再往上取就超出右端点范围了）
为了分清这两种情况，我们引入了limit标记：

若当前位limit=1而且已经取到了能取到的最高位时，下一位limit=1；
若当前位limit=1但是没有取到能取到的最高位时，下一位limit=0；
若当前位limit=0时，下一位limit=0。
我们设这一位的标记为limit，这一位能取到的最大值为res，则下一位的标记就是i==res&&limit（i枚举这一位填的数）

dp值的记录和使用
最后我们考虑dp数组下标记录的值

本文介绍数位dp是在记忆化搜索的框架下进行的，每当找到一种情况我们就可以这种情况记录下来，等到搜到后面遇到相同的情况时直接使用当前记录的值。

dp数组的下标表示的是一种状态，只要当前的状态和之前搜过的某个状态完全一样，我们就可以直接返回原来已经记录下来的dp值。

再举个例子

假如我们找[0,123456]中符合某些条件的数

假如当我们搜到1000??时，dfs从下返上来的数值就是当前位是第1位，前一位是0时的方案种数，搜完这位会向上，这是我们可以记录一下：当前位第1位，前一位是0时，有这么多种方案种数

当我们继续搜到1010??时，我们发现当前状态又是搜到了第1位，并且上一位也是0，这与我们之前记录的情况相同，这样我们就可以不继续向下搜，直接把上次的dp值返回就行了。

注意，我们返回的dp值必须和当前处于完全一样的状态，这就是为什么dp数组下标要记录pos,pre等参量了。

最重要的来了————————————————————

接着上面的例子，范围[0,123456]

如果我们搜到了1234??，我们能不能直接返回之前记录的：当前第1位，前一位是4时的dp值？

答案是否定的

我们发现，这个状态的dp值被记录时，当前位也就是第1位的取值是[0,9]，而这次当前位的取值是[0,5]，方案数一定比之前记录的dp值要小。

当前位的取值范围为什么会和原来不一样呢？

如果你联想到了之前所讲的知识，你会发现：现在的limit=1，最高位有取值的限制。

因此我们可以得到一个结论：当limit=1时，不能记录和取用dp值！

没有限制的情况占多数，所以只记录没有高位限制的情况
if(!limit)
{
dp[pos][pre] = ans;
}

有limit=1限制的怎dp[pos][pre]么办呢？每次都重新算。

前导0标记lead

由于我们要搜的数可能很长，所以我们的直接最高位搜起

举个例子：假如我们要从[0,1000]找任意相邻两数相等的数

显然111,222,888等等是符合题意的数

但是我们发现右端点1000是四位数

因此我们搜索的起点是0000，而三位数的记录都是0111,0222,0888等等

而这种情况下如果我们直接找相邻位相等则0000符合题意而0111,0222,0888都不符合题意了

所以我们要加一个前导0标记

如果当前位lead=1而且当前位也是0，那么当前位也是前导0，pos-1继续搜；
如果当前位lead=1但当前位不是0，则本位作为当前数的最高位，pos-1继续搜；（注意这次根据题意st或其他参数可能发生变化）

当然前导0有时候是不需要判断的，上述的例子是一个有关数字结构上的性质，0会影响数字的结构，所以必须判断前导0；而如果我们研究的是数字的组成（例如这个数字有多少个111之类的问题），0并不影响我们的判断，这样就不需要前导0标记了。总之，这个因题而异，并不是必须要标记（当然记了肯定是不会出错的）

类似上述的分析过程，我们也可以得出：当lead=1时，也不能记录和取用dp值！

if(!limit&&!lead) dp[pos][pre]=ans;

前导0是无效的不用管，比如说0001含前导0就可以直接视作1，1001就不含前导0

ybt

通过

测试点	结果	内存	时间
测试点1	答案正确	604KB	1MS
测试点2	答案正确	596KB	2MS
测试点3	答案正确	604KB	2MS
测试点4	答案正确	600KB	1MS
测试点5	答案正确	596KB	2MS
测试点6	答案正确	600KB	1MS
测试点7	答案正确	604KB	1MS
测试点8	答案正确	604KB	1MS
测试点9	答案正确	596KB	2MS
测试点10	答案正确	604KB	1MS

LOJ

LUOGU

AC代码如下：

#include <bits/stdc++.h>
using namespace std;
int num[12],dp[12][12];
int dfs(int pos,int pre,int limit,int lead){
	int ans=0,i,up;
	if(pos==-1)return 1;
	if(!limit&&dp[pos][pre]!=-1&&!lead)return dp[pos][pre];
	up=limit?num[pos]:9;
	for(i=0;i<=up;i++){
		if(lead){//有前导0不受限制 
			ans+=dfs(pos-1,i,limit&&i==up,lead&&i==0);
		}else if(i-pre>=2||i-pre<=-2)//无前导0受限 
			ans+=dfs(pos-1,i,limit&&i==up,lead&&i==0);
	}
	if(!limit&&!lead)dp[pos][pre]=ans;
	return ans;
}
int solve(int x){
	int pos=0;
	while(x){
		num[pos++]=x%10;
		x/=10;
	}
	return dfs(pos-1,-1,1,1);
}
int main(){
	int lt,rt;
	scanf("%d%d",&lt,&rt);
	memset(dp,-1,sizeof(dp));
	printf("%d\n",solve(rt)-solve(lt-1));
	return 0;
}