【NOI 2016模拟6.16】gene

题目描述

你有$n$个数$\{ a_1, \cdots, a_n\}$，以及$m$个数$\{b_1, \cdots, b_m\}$

对于任意一个$a_i$，你可以任意选择任意多个$a_i^{b_i}$乘起来对$p$取模，得到若干个数。
问所有这样子得到的数的并集大小。

$n\leq 10^4, m\leq 10^5, 1<p\leq10^9$，且$p$是个质数。

---

分析

首先$a_i^{-b_j}$是会被取到的，只需要取循环节的上一个就可以了。

那么根据裴蜀定理，$\forall x\in[0,p-1), (b_0, b_1, \cdots, b_m, p-1)|x$都可以被取到。

考虑到如此多个$a_i$取并集不好做，那么求出离散对数以后，这个底数都会变成原根。问题就放到指数上了。

也就是说，我们记$(ind(a_i), b_1, \cdots, b_m, p-1)=c_i$
考虑怎么计算这个$c_i$。根据最大公约数的性质，这个东西等价于$( ind( a_i ), ( b_1, \cdots, b_m, p-1 ))$
$ind( x )$可以用离散对数算出来，但是考虑到这里我们需要求它和别的东西的最大公约数，我们没必要真的算出离散对数。

考虑$x$的阶$ord( x )$，根据定义，$ind( x )ord( x )\equiv 0 \pmod{ \varphi(p) }$，也就是说$ind( x )=\frac{k\varphi( p ) }{ord( x )}$，且$ord( x )$与$k$互质。

考虑费马小定理$a^{\varphi( p )}\equiv 1 \pmod{p}$，那么$ord( x )$必定是$\varphi(p)$的约数。

下证$( ind( x ), \varphi( p ) )=( \frac{ \varphi( p）}{ ord( x ) }, \varphi( p ))=\frac{\varphi(p)}{ord( x )}$

左式可以如下变换。

$$( ind( x ), \varphi( p ) )=\frac{ \left( ind( x )ord( x ), ord( x )\varphi( p ) \right ) }{ord( x )}=\frac{ \left( k\varphi(p), ord( x )\varphi( p ) \right ) }{ord( x )}=\frac{ \varphi(p)\left( k, ord( x ) \right ) }{ord( x )}$$

由于$( k, ord( x ) )=1$，故左式等于右式。

有了这个以后，我们就可以在这个问题里将阶来取代离散对数起到相同的作用。

之后的问题就是给定$n$个$c_i$，问有点多少个数至少是一个$c_i$的倍数。
形式化地说，我们需要统计有多少个$x$，满足

$$x\in[0, \varphi(p)), \exists i, c_i|x$$

又因为$c_i|n$，故$c_i|( n, x )$

那么我们就可以枚举最大公约数，剩下的式子就是

$$\sum_{d=1}^n [\exists i, c_i|d]\varphi(\frac{n}{d})$$

时间复杂度$O( m\log b_i + \sigma_0(p)n\log p +n\ln n)$
空间复杂度$O(n)$