【洛谷 P3281】 [SCOI2013]数数

题目描述

Fish 是一条生活在海里的鱼，有一天他很无聊，就开始数数玩。他数数玩的具体规则是：

确定数数的进制B

确定一个数数的区间[L, R]

对于[L, R] 间的每一个数，把该数视为一个字符串，列出该字符串的每一个（连续的）子串对应的B进制数的值。

对所有列出的数求和。现在Fish 数了一遍数，但是不确定自己的结果是否正确了。由于[L, R] 较大，他没有多余精力去验证是否正确，你能写一个程序来帮他验证吗？

输入格式

输入包含三行。

第一行仅有一个数B，表示数数的进制。

第二行有N +1 个数，第一个数为N，表示数L 在B 进制下的长度为N，接下里的N个数从高位到低位的表示数L 的具体每一位。

第三行有M+ 1 个数，第一个数为M，表示数R 在B 进制下的长度为M，接下里的M个数从高位到低位的表示数R 的具体每一位。

输出格式

输出仅一行，即按照Fish 数数规则的结果，结果用10 进制表示，由于该数可能很大，输出该数模上20130427的模数。

输入输出样例

输入 #1 复制
10
3 1 0 3
3 1 0 3
输出 #1 复制
120

说明/提示

【样例解释】

[103, 103] 之间仅有数103，该数的所有子串包括1, 10, 103, 0, 03, 3，其和为120。

【数据范围与约定】

20% 数据，0 <= R <= L <= 10^5。

50% 数据，2 <= B <= 1000，1 <= N,M <= 1000。

100% 数据，2 <= B <= 10^5，1 <= N,M <= 10^5。

思路

首先我们考虑解决这样一个问题，求[1,x-1)内所有数的答案

把x写成B进制形式 x1, x2, x3, … , xn。

首先考虑不足n位的数

然后对于达到了n位的数，枚举第一个小于xi的数位。

由于是开区间，必然存在这样一个数位。

这种方法可以减少前导零的分类，还是比较方便的

下面讲一下这道题的DP

这道题是融合了很浓厚的数学思想的

下文设P[i] = 10^i，S[i]为P[i]前缀和

考虑给定一个数x1, x2, …, xn，求它的子串和。

考虑每个数位的贡献，第i个数位的贡献是

i * xi * S[n - i]
理解一下这个式子

定义d[i][0]表示还要填i位，所有前缀子串和，d[i][1]表示还要填i位，所有子串和。

这个转移还是比较显然的

对于不足n位的数，显然直接枚举最高位就行了

达到n位的数，我们需要考虑三段的贡献，即：第一个小于xi的数位，第一个小于xi的数位之前，第一个小于xi的数位之后

具体情况可以看一下下面的代码

然而我们还有一个问题没有解决

上面的做法是O(nB)的。

只要把枚举该位填了什么改成通过

1 + 2 + … + n = n * (n + 1) / 1
优化一下即可

代码的SV表示原数x的一段前缀在整个数上的贡献

千万注意每次DP前清零SV。

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int mod = 20130427;
int cnt, a[100005];
int B, SV[100005], d[100005][2], P[100005], S[100005];
void read()
{
	memset(SV, 0, sizeof(SV));
	for(int i = cnt - 1; i >= 0; i--)
		scanf("%lld", &a[i]), SV[i] = (SV[i + 1] + a[i] * (cnt - i) % mod * S[i] % mod) % mod;
}
int calc(int x, bool p)
{
	if(x <= 0) return 0;
	if(d[x][p] != -1) return d[x][p];
	if(p)
		return d[x][1] = (B * calc(x - 1, 1) % mod + (B * (B - 1) / 2) % mod * P[x - 1] % mod * S[x - 1] % mod + B * calc(x - 1, 0) % mod) % mod;
	else
		return d[x][0] = (B * calc(x - 1, 0) % mod + (B * (B - 1) / 2) % mod * P[x - 1] % mod * S[x - 1] % mod) % mod;
}
int solve()
{
	int i, ans = 0;
	for(i = 1; i < cnt; i++)
		ans = (ans + (B - 1) * calc(i - 1, 1) % mod + (B * (B - 1) / 2) % mod * P[i - 1] % mod * S[i - 1] % mod + (B - 1) * calc(i - 1, 0) % mod) % mod;
	for(i = cnt - 1; i >= 0; i--)
		if(i == cnt - 1)
			ans = (ans + (a[i] * (a[i] - 1) / 2) % mod * P[i] % mod * S[i] % mod + (a[i] - 1) * calc(i, 1) % mod + (a[i] - 1) * calc(i, 0) % mod) % mod;
		else
			ans = (ans + SV[i + 1] * a[i] % mod * P[i] % mod + (a[i] * (a[i] - 1) / 2) % mod * P[i] % mod * S[i] % mod * (cnt - i) % mod + a[i] * calc(i, 1) % mod + a[i] * calc(i, 0) % mod * (cnt - i) % mod) % mod;
	return ans;
}
int count()
{
	return SV[0];
}
signed main()
{
	int a1, a2, c1, c2;
	memset(d, -1, sizeof(d));
	scanf("%lld%lld", &B, &cnt);
	P[0] = S[0] = 1;
	for(int i = 1; i <= 100003; i++)
		P[i] = P[i - 1] * B % mod, S[i] = (S[i - 1] + P[i]) % mod;
	read();
	c1 = count();
	a1 = solve();
	scanf("%lld", &cnt);
	read();
	c2 = count();
	a2 = solve();
	printf("%lld\n", (a2 - a1 + c2 + mod * mod) % mod);
	return 0;
}