Primes and Queries - Google Kickstart 2021 Round D

Primes and Queries - Google Kickstart 2021 Round D第四题

You are given a prime number P.
Let’s define V(x) as the degree of P in the prime factorization of x. To be clearer, if V(x)=y then x is divisible by $P^y$, but not divisible by $P^{y+1}$.
Also we define V(0)=0.
For example, when P=3, and x=45, since 45=5⋅32, therefore V(45)=2.
You are also given an array A with N elements. You need to process Q queries of 2 types on this array:
type 1 query: 1 pos val - assign a value val to the element at pos, i.e. $A_{pos}=val$
type 2 query: 2 S L R - print ${\sum }_{i=L}^{R}V(A_i^S-(A_i(\mathrm{m}\mathrm{o}\mathrm{d}P){)}^S)$.

题目大意

给定一个素数$P$,定义$V(x)$为$P$在$x$质因子分解里的度数（比如质数是2，20可以被4整除，不能被8整除，20的度数就是2），定义$V(0)=0$。 现在给定一个长度为$N$的序列($1\le N\le 5\ast 10^5$)，试处理$Q$个操作($1\le Q\le 10^5$)。操作有如下两种：
1.$1posval$:把第$pos$个数改为$val$ ;
2. $2SLR$:输出${\sum }_{i=L}^{R}V(A_i^S-(A_i(\mathrm{m}\mathrm{o}\mathrm{d}P){)}^S)$.

思路分析

很明显是一道线段树问题，修改就不说了，区间求和只是看上去复杂一点。为了方便我们令$B=A(\mathrm{m}\mathrm{o}\mathrm{d}P)$,那么$A_i^S-(A_i(\mathrm{m}\mathrm{o}\mathrm{d}P){)}^S=A_i^S-B_i^S$
注意到$A_i$和$B_i$模$P$同余，我们分情况讨论：

1.$P|A_i$：每一项的贡献$V(A_i^S-B_i^S)=V(A_i-B_i)+V(S\ast A_i^S)=V(A_i-B_i)+S\ast V(A_i^S)$。
2.$P\nmid A_i$： 我们用提升指数引理：如果P是2且$S$为偶数，那么$V(A_i^S-B_i^S)=V(A_i-B_i)+V(A_i+B_i)+V(S)-1$,否则$V(A_i^S-B_i^S)=V(A_i-B_i)+V(S\ast A_i^S)=V(A_i-B_i)+V(S)$。

$V$这个函数的复杂度上界只有$log(n)$,所以我们每次可以直接暴力计算。我们建立两颗线段树，一颗上面是所有当前能被整除的项，一颗是所有不能被整除的项。修改时，如果被修改的数从不能被整除变成可以被整除，只需要把第二颗树上的当前位置归零，再在第一棵树上修改即可，反之亦然。求和时，把两棵树的结果加起来就可以了。注意一下如果质数是2的话，那么我们第二颗树上记录的应当是$V(A_i-B_i)+V(A_i+B_i)$,小细节不要忘记。

分析一下时间复杂度：就是一般线段树的时间复杂度，外加建树时需要额外预处理$V$函数的值，总复杂度是$O(Nlog(A_i)+Q(logN+logS))$。

题目比较水，可以算是纯数学题了，没什么技巧可言，希望以后的题目能更有新意。