CF628D 题解

题目传送门

数位 DP 题,较为套路。

定义状态 \(\mathrm{dp[pos][res]}\) 表示当前填到第 \(\mathrm{pos}\) 位,模 \(m\) 余数为 \(\mathrm{res}\),满足题意的数的个数。当 \(\mathrm{pos=0}\) 且 \(\mathrm{res=0}\) 时,计入总数。

由于记忆化搜索时是从高位往低位填,但对应的下标为 \(\mathrm{len}\) 到 \(1\),所以第 \(\mathrm{pos}\) 位从题目角度看应为第 len-pos+1 位。

Q :\([l,r]\) 范围内的数字并不能存在前导零,否则奇偶位数无法判断。故需要判断前导零,这样状态中就应该再加一维,变成 \(\mathrm{dp[pos][len][res]}\),才能防止状态的重复。但这样空间就到达了 \(\mathrm{O(2000^3)}\),必定会 MLE。

A :确实如此。但是题目中提到了:保证 \(l,r\) 位数相同。因此如果我们不管前导零,那么那些不合法的状态的数,去除前导零后的位数一定是比 \(l,r\) 的位数都要小的,而这些数在统计 \(l\) 和 \(r\) 的时候都会被计算,相减后就会被抵消,因此不合法的状态都会被剔除,无需考虑前导零。

根据题目奇偶位限制,对应写出一下代码:

for(int i=0;i<=up;i++)
{
	int wei=(len-pos+1);
	ll tmp=0;
	if(wei&1)
	{
		if(i!=d)tmp=dfs(pos-1,len,(res*10+i)%m,limit&&i==up);
	}
	else
	{
		if(i==d)tmp=dfs(pos-1,len,(res*10+i)%m,limit&&i==up);
	}
	ans=(ans+tmp)%mod;
}

注意到数据范围:\(1\le l\le r\le10^{2000}\)。因此 \(l\) 和 \(r\) 是以字符串形式读入的。普通的 solve(r)-solve(l-1) 因为 l-1 的难以实现而行不通。

因此我们需要单独判断 \(l\) 是否满足条件,直接模拟即可。注意位数与读入时下标间的不同。

ll solve_for_l()
{
	int len=strlen(l),res=0;
	for(int i=0;i<len;i++)
	{
		int ss=(l[i]^48);
		if((i+1)&1)
		{
			if(ss==d)return 0;
		}
		else
		{
			if(ss!=d)return 0;
		}
		res=(res*10+ss)%m;
	}
	return res==0;
}

因此答案即为 solve(r)-solve(l)+solve_for_l(),注意取模细节。

上一篇:第五章 相似矩阵及二次型


下一篇:JS高程12.2.3元素大小的学习笔记