Description
Solution
算法一
注意到 [ 1 , 1 0 4 ] [1,10^4] [1,104] 中幸运数字只有 30 30 30 个左右——对于一次询问,枚举幸运数字,计算出它的出现次数并求和。
注意到这是一个形如“区间修改,区间查询某种值出现次数”的题,可以用树套树维护。
总复杂度 O ( 30 n log 2 n ) O(30 n \log^2 n) O(30nlog2n),常数极大,显然无法通过。
算法二
令 f i f_i fi 表示,当前 a i a_i ai 到离它最近的一个幸运数字的距离。
对于一次修改,
[
l
,
r
]
[l,r]
[l,r] 中的所有满足
f
i
≥
d
f_i \ge d
fi≥d 的
a
i
a_i
ai 减去了
d
d
d,其他位置如何变化并不确定。
对于一次查询,答案即为
f
[
l
,
r
]
f[l,r]
f[l,r] 中
0
0
0 出现的次数。
考虑用线段树维护。具体来说,对于线段树上的每个节点,维护 最小 f f f 与 0 0 0 出现的次数。对于一次修改,找到对应的 log n \log n logn 个区间,若某个区间的最小值不小于 g g g 就直接打标记返回,否则暴力向两个孩子递归,重构子树;对于一次查询,直接将各个区间中 0 0 0 出现的次数加起来即可。
总复杂度 O ( 30 n log n ) O(30 n \log n) O(30nlogn),常数较小,可以通过。