2023.01.26 数论 学习笔记
讲课人:罗恺。
学习/复习了很多数论算法。
有些题没有分析复杂度,要么是不会分析,要么是分析出来全是常数。
Pohlig-Hellman
问题:给定 \(a\),质数 \(p\) 和 \(p\) 的一个原根 \(g\),求 \(x\) 使得 \(g^x\equiv a\pmod p\)。
不妨 \(0\le a<p\)。特判掉 \(a=0,1\),只需考虑 \(2\le a< p\) 的情况。
直接 BSGS 的复杂度是 \(O(\sqrt p)\) 的,如果 \(p\) 达到了 \(10^{18}\) 就没救了。此时如果 \(p\) 满足一些性质,例如 \(p-1\) 的所有质因子都比较小,就可以用 Pohlig-Hellman 来求解。
我们先把 \(p-1\) 分解质因数,设 \(p-1=p_1^{\alpha_1}p_2^{\alpha_2}\cdots p_k^{\alpha_k}\)。假设我们能求出 \(x\) 模 \(p_1^{\alpha_1}\) 的余数,那么我们也可以求出 \(x\) 模 \(p_2^{\alpha_2}\sim p_k^{\alpha_k}\) 的余数,从而用 CRT 把 \(x\) 求出。
考虑怎么求 \(x\) 模 \(p_1^{\alpha_1}\) 的余数。将 \(x\) 用 \(p_1\) 进制表出,即设 \(x=\sum_{i=0}^{\alpha_1-1}p_1^ia_i\cdots(1)\),其中 \(0\le a_i\le p_1-1\)。由于原问题在 \([0,p-1]\) 中有唯一解,因此 \(a_i\) 存在且唯一。
假设我们已经求出了 \(a_0\sim a_{r-1}\),考虑怎么求 \(a_r\)。我们已知 \(g^x\equiv a\pmod p\),那么 \(g^{x\frac{p-1}{p_1^{r+1}}}\equiv a^{\frac{p-1}{p_1^{r+1}}}\pmod p\),代入 \((1)\) 式得,\(g^{\sum_{i=0}^{\alpha_1-1}p_1^ia_i\frac{p-1}{p_1^{r+1}}}\equiv a^{\frac{p-1}{p_1^{r+1}}}\pmod p\),我们发现当 \(i>r\) 时左式指数一定是 \(p-1\) 的倍数,而 \(g^{p-1}\equiv 1\pmod p\),因此左式仅需考虑 \(i\le r\) 时的指数。
因此有 \(g^{\sum_{i=0}^{r-1}p_1^ia_i\frac{p-1}{p_1^{r+1}}}g^{a_r\frac{p-1}{p_1}}\equiv a^{\frac{p-1}{p_1^{r+1}}}\pmod p\),\(O(p_1)\) 枚举 \(a_r\) 并检验是否合法即可。
总时间复杂度 \(O(\sum_{i=1}^k \alpha_i(p_i+\log a))\)。
应用:HDU6632 discrete logarithm problem。
首先找到 \(p\) 的原根 \(g\)。可以用 Pohlig-Hellman 求出 \(a=g^{a_1},b=g^{b_1}\),于是问题变为找 \(x\) 使得 \(g^{a_1x}\equiv g^{b_1}\pmod p\),即 \(a_1x\equiv b_1\pmod{p-1}\),可以用 exgcd 或快速幂求解。
Min_25 筛
能用 Min_25 筛算前缀和的函数 \(f(x)\) 一般需满足以下性质:对于质数 \(p\),\(f(p)\) 是一个关于 \(p\) 的项数较少的多项式,且 \(f(p^c)\) 可以快速计算。
整个过程分为两部分:1. 对所有 \(x\in\{\lfloor\frac{n}{d}\rfloor|1\le d\le n\}\) 求出不超过 \(x\) 的所有质数的 \(f\) 之和。2. 对所有 \(x\in\{\lfloor\frac{n}{d}\rfloor|1\le d\le n\}\) 求出不超过 \(x\) 的所有数的 \(f\) 之和。
第一部分:
由于对于质数 \(p\),\(f(p)\) 是一个关于 \(p\) 的项数较少的多项式,所以我们可以把它拆成若干个单项式的和,最后再把每一项的结果加起来就可以了。
现在我们考虑怎么求出 \(g(p)=p^k\) 的前缀和。
设 \(p_i\) 为第 \(i\) 小的质数,如 \(p_1=2,p_2=3\),特别地,认为所有质数都 \(>p_0\)(尽管 \(p_0\) 无意义)。设不超过 \(n\) 的质数有 \(t_n\) 个。令 \(sum_i\) 等于前 \(i\) 小的质数的 \(g\) 之和。令 \(G(n,x)\) 等于 \(n\) 以内的所有质数以及所有最小质因子大于 \(p_x\) 的合数的 \(g\) 之和。
首先 \(G(n,0)=\sum_{i=2}^n g(i)\) 可以快速计算。如果我们已知 \(G(n,x-1)\),怎么求 \(G(n,x)\)?可以发现,\(G(n,x-1)\) 和 \(G(n,x)\) 相比多了 \(n\) 以内最小质因子等于 \(p_x\) 的合数的 \(g\) 之和。
因此,如果 \(p_x^2>n\),则 \(G(n,x)=G(n,x-1)\);否则,因为 \(g\) 是完全积性函数,所以 \(G(n,x)=G(n,x-1)-g(p_x)(G(\lfloor\frac{n}{p_x}\rfloor,x-1)-sum_{x-1})\)。
关于 \(G\),其第一维一定形如 \(\lfloor\frac{n}{d}\rfloor(1\le d\le n)\) 的形式。关于 \(sum\),只需预处理前 \(t_{\lfloor\sqrt n\rfloor}\) 项。
不超过 \(n\) 的所有质数的 \(g\) 之和就是 \(G(n,t_n)\)。进而可以求出不超过 \(n\) 的所有质数的 \(f\) 之和 \(G_n\)。
第二部分:
令 \(F(n,x)\) 等于 \(n\) 以内的所有大于 \(p_x\) 的质数以及所有最小质因子大于 \(p_x\) 的合数的 \(f\) 之和。不妨 \(p_x\le n\)。
对于质数的部分,其贡献为 \(G_n-sum_{x}\)。
对于合数的部分,枚举最小质因子 \(p_i(i>x)\) 及其出现次数 \(e\),则贡献为 \(f(p_i^e)(F(\lfloor\frac{n}{p_i^e}\rfloor,i)+[e>1])\)。
原问题的答案就是 \(F(n,0)+f(1)\)。直接递归求解,不用记忆化。
总时间复杂度据说可以被证明是 \(O(n^{1-\epsilon})\) 或者 \(O(\frac{n^{\frac{3}{4}}}{\log n})\) 的。
1.P3726 [AH2017/HNOI2017] 抛硬币
首先让 A 先抛 \(a-b\) 次,枚举 A 在这 \(a-b\) 次中正面朝上的次数 \(y\),则方案数为 \(\binom{a-b}{y}\)。接下来 A 和 B 要一起抛 \(b\) 次,A 要获胜等价于在这 \(b\) 次中 A 正面朝上的次数减 B 正面朝上的次数 \(\ge 1-y\)。如果是要求 A 正面朝上的次数减 B 正面朝上的次数恰好等于某个数,则可以枚举 A、B 正面朝上的次数,用组合数表示,可以用范德蒙德卷积进行优化,由于模数不是质数,所以这部分的时间复杂度取决于 exLucas 的复杂度,即 \(O(2^k+5^k)\) 预处理,\(O(\log a)\) 单次查询。同时注意到 \(y=1\) 时是很好算的,且知道 \(y\) 时的答案可以通过再计算一个组合数推出 \(y-1\) 或 \(y+1\) 时的答案,且 \(0\le y\le a-b\),所以一共要算 \(O(a-b)\) 个组合数。
总时间复杂度 \(O(2^k+5^k+(a-b)\log a)\),空间复杂度 \(O(5^k)\)。
2.CF986F Oppa Funcan Style Remastered
首先只可能用 \(k\) 的质因子。
如果 \(k\) 只有一个质因子,那做完了。
如果 \(k\) 只有两个质因子,把式子列出来后可以 exgcd 判。
如果 \(k\) 有至少三个质因子,则最小的质因子不超过 \(O(k^{\frac{1}{3}})\),可以同余最短路。
设有 \(t_k\) 个不同的 \(k\),则时间复杂度为 \(O(t\log k+\sqrt k+t_k(\frac{\sqrt k}{\ln k}+k^{\frac{1}{3}}\log^2k))\),空间复杂度 \(O(t+\frac{\sqrt k}{\omega}+\frac{\sqrt k}{\ln k}+k^{\frac{1}{3}})\)。
3.CF571E Geometric Progressions
设在 \(a,b\) 中出现过的质因子有 \(t\) 个,分别为 \(p_1\sim p_t\)。设 \(a_i=\prod_{j=1}^t p_j^{c_{i,j}},b_i=\prod_{j=1}^t p_j^{d_{i,j}}\)。假如有解,不妨设其中一组解是 \(a_1b_1^{x_1}=\cdots=a_nb_n^{x_n}\)。
\(t=1\) 时可以 exCRT 求解,\(n=1\) 时可以特判。不妨设 \(t,n\ge 2\)。
如果存在 \(i,j\) 使得 \(d_{i,j}=0\),考虑所有 \(d_{x,j}(1\le x\le n)\)。如果 \(d_{x,j}\) 全都为 \(0\),看 \(c_{x,j}\) 是否全部相等,如果全部相等则可以忽略质因子 \(p_j\),否则无解;如果 \(d_{x,j}\) 中有不为 \(0\) 的,那么用 \(d_{i,j}=0\) 和 \(d_{x,j}\ne 0\) 的列个方程可以发现最多有一个可行解,验证下是否合法即可。
现在所有剩余的 \(d\) 都不为 \(0\) 了。如果有解,考虑 \(j\in[2,t]\),要满足对于 \(2\le i\le n,\),都有 \(c_{1,1}+x_1d_{1,1}=c_{i,1}+x_id_{i,1},c_{1,j}+x_1d_{1,j}=c_{i,j}+x_id_{i,j}\),移项得 \(x_1d_{1,1}-x_id_{i,1}=c_{i,1}-c_{1,1},x_1d_{1,j}-x_id_{i,j}=c_{i,j}-c_{1,j}\)。如果存在 \(i\) 满足 \(\frac{d_{1,1}}{d_{1,j}}\ne \frac{d_{i,1}}{d_{i,j}}\),则至多有一个可行解,检验即可;否则,如果存在 \(i\) 使得 \(\frac{d_{1,1}}{d_{1,j}}\ne\frac{c_{i,1}-c_{1,1}}{c_{i,j}-c_{1,j}}\),则无解;否则,可以忽略质因子 \(p_j\)。
重复上述过程直至 \(t=1\) 为止即可。
时间复杂度 \(O(\sqrt a+n\frac{\sqrt a}{\ln a}+n^2\log a)\),空间复杂度 \(O(\frac{\sqrt a}{\omega}+\frac{a}{\ln a}+n^2\log a)\)。
4.UOJ#188. 【UR #13】Sanrd
类似 Min_25 筛。
差分后,原问题等价于求 \(1\sim n\) 的次大质因子之和。
设 \(F(n,x)\) 等于 \(n\) 以内的所有最小质因子大于 \(p_x\) 的合数的次大质因子之和。设不超过 \(n\) 的质数有 \(t_n\) 个。
枚举合数的最小质因子 \(p_i(i>x)\) 及其出现次数 \(e\),则该合数除去 \(p_i^e\) 后可能为合数、质数、或者 \(1\),故有转移 \(F(n,x)=\sum_{i>x,p_i^2\le n}\sum_{e\ge 1}(F(\lfloor\frac{n}{p_i^e}\rfloor,i)+(\max(t_{\lfloor\frac{n}{p_i^e}\rfloor}-x,0)+[e>1])p_i)\)。
同样的 \(t\) 只有下标为 \(\lfloor\frac{n}{d}\rfloor(1\le d\le n)\) 的位置的值有用,\(F\) 第一维也只可能是 \(\lfloor\frac{n}{d}\rfloor\) 的形式。根据前文 Min_25 筛的部分,可以看出求出所有需要的 \(t\) 就是第一部分,求 \(F\) 就类似于第二部分。
问题的答案就是 \(F(n,0)\)。直接递归求解,不用记忆化。
时间复杂度 \(O(r^{1-\epsilon})\) 或 \(O(\frac{r^{\frac{3}{4}}}{\log r})\),空间复杂度 \(O(\sqrt r)\)。
5.P5172 Sum
首先特判掉 \(r\) 是完全平方数的情况。
我们发现,\((-1)^a=1-2(a\bmod 2)=1-2(a-2\lfloor\frac{a}{2}\rfloor)\),因此原式等于 \(n-2\sum_{i=1}^n \lfloor i\sqrt r\rfloor+4\sum_{i=1}^n \lfloor \frac{i\sqrt r}{2}\rfloor\)。
令 \(t=\sqrt r\),令 \(f(n,a,b,c)=\sum_{i=1}^n \lfloor i\frac{at+b}{c}\rfloor\),答案即为 \(n-2f(n,1,0,1)+4f(n,1,0,2)\)。
考虑怎么求 \(f(n,a,b,c)\),要求 \(a,c\ne 0\)。如果 \(\frac{at+b}{c}\) 不在 \([0,1)\) 间,则可以把 \(\frac{at+b}{c}\) 的整数部分先提出来(即将 \(b\) 减去某个值),使得 \(\frac{at+b}{c}\) 在 \([0,1)\) 间。
\(f(n,a,b,c)=\sum_{i=1}^n \lfloor i\frac{at+b}{c}\rfloor=\sum_{i=1}^n\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}[\lfloor i\frac{at+b}{c}\rfloor\ge j]=\sum_{i=1}^n\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}[i\frac{at+b}{c}\ge j]\),由于 \(a\ne0\) 且 \(r\) 不为完全平方数,所以不等号两边不会相等,因此 \(f(n,a,b,c)=\sum_{i=1}^n\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}[i\frac{at+b}{c}\gt j]=\sum_{i=1}^n\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}[j\frac{c}{at+b}<i]=\sum_{i=1}^n\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}[j\frac{c(at-b)}{(at+b)(at-b)}<i]=\sum_{i=1}^n\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}[j\frac{c(at-b)}{a^2r-b^2}<i]=\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}\sum_{i=1}^n[j\frac{c(at-b)}{a^2r-b^2}<i]=\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}\sum_{i=1}^n[\lfloor j\frac{c(at-b)}{a^2r-b^2}\rfloor<i]=\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}(n-\lfloor j\frac{c(at-b)}{a^2r-b^2}\rfloor)=n\lfloor n\frac{at+b}{c}\rfloor-\sum_{j=1}^{\lfloor n\frac{at+b}{c}\rfloor}\lfloor j\frac{c(at-b)}{a^2r-b^2}\rfloor=n\lfloor n\frac{at+b}{c}\rfloor-f(\lfloor n\frac{at+b}{c}\rfloor,ca,-cb,a^2r-b^2)\)。由于 \(r\) 不为完全平方数,所以 \(a^2r\) 不为完全平方数,所以 \(a^2r-b^2\ne 0\);由于原先 \(a,c\ne 0\),所以现在 \(ca\ne 0\),因此 \(f(\lfloor n\frac{at+b}{c}\rfloor,ca,-cb,a^2r-b^2)\) 仍满足我们的要求,可以递归求解。由于 \(n>0\) 时每递归一次 \(n\) 至少减少 \(1\),\(n=0\) 时答案为 \(0\),所以可以在有限步内得到答案。
上述做法的正确性是显然的。但是还有两个问题,一是 \(f\) 的递归层数是多少(这会影响时间复杂度),二是在递归的过程中 \(f\) 的四维参数值域会有多大(这会影响用什么类型的变量存储)。
先分析前者。假设递归到 \(f(n,a,b,c)\) 时把 \(\frac{at+b}{c}\) 调整到 \([0,1)\) 间后 \(a,b,c\) 分别变成了 \(a',b',c'\),那么下一层递归就是 \(f(\lfloor n\frac{a't+b'}{c'}\rfloor,c'a',-c'b',a'^2r-b'^2)\),令 \(A=c'a',B=-c'b',C=a'^2r-b'^2\),即递归到 \(f(\lfloor n\frac{a't+b'}{c'}\rfloor,A,B,C)\)。设递归到 \(f(\lfloor n\frac{a't+b'}{c'}\rfloor,A,B,C)\) 时把 \(\frac{At+B}{C}\) 调整到 \([0,1)\) 间后 \(A,B,C\) 分别变成了 \(A',B',C'\),那么再下一层递归就是 \(f(\lfloor\lfloor n\frac{a't+b'}{c'}\rfloor\frac{A't+B'}{C'}\rfloor,C'A',-C'B',A'^2r-B'^2)\)。可以发现,\(\frac{A't+B'}{C'}\) 其实就是 \(\frac{At+B}{C}\) 的小数部分(\(x\) 的小数部分 \(\{x\}\) 定义为 \(x-\lfloor x\rfloor\)),而 \(\frac{At+B}{C}=\frac{c'a't-c'b'}{a'^2r-b'^2}=\frac{c'(a't-b')}{(a't-b')(a't+b')}=\frac{c'}{a't+b'}\),因此 \(f(n,a,b,c)\) 递归两层后变成 \(f(\lfloor\lfloor n\frac{a't+b'}{c'}\rfloor\{\frac{c'}{a't+b'}\}\rfloor,C'A',-C'B',A'^2r-B'^2)\)。令 \(n'=\lfloor\lfloor n\frac{a't+b'}{c'}\rfloor\{\frac{c'}{a't+b'}\}\rfloor\),令 \(x=\frac{a't+b'}{c'}(x<1)\),则 \(n'=\lfloor\lfloor nx\rfloor\{\frac{1}{x}\}\rfloor\le nx\{\frac{1}{x}\}\),不难发现 \(nx\{\frac{1}{x}\}\) 取到最大值时 \(1\lt \frac{1}{x}\lt 2\),故设 \(\frac{1}{x}=1+y\),其中 \(0<y<1\),则 \(nx\{\frac{1}{x}\}\le n\frac{y}{1+y}\),可以发现 \(\frac{y}{1+y}<\frac{1}{2}\),故 \(n'\le n\frac{y}{1+y}\lt \frac{n}{2}\)。也就是说,每经过两次递归,\(n\) 就至少除以 \(2\),那么递归层数就是 \(O(\log n)\) 的。
对于后者,我们要在递归计算 \(f\) 时加一个优化:把 \(a,b,c\) 同时除以 \(\gcd(a,b,c)\)。根据定义这显然是对的,并且可以在一定程度上减小 \(a,b,c\) 的值域。下面我们说明,在题目给定数据范围内,加上这个优化后,只要用 long long 存储就可以了。
首先,可以发现 \(f\) 在递归的过程中 \(n\) 会越变越小,除此之外,\(a,b,c\) 三维的变化不受 \(n\) 的影响,只受它们本身的值和 \(r\) 的影响,\(n\) 只起到了控制递归轮数的作用,并且 \(n\) 越大递归轮数越多。所以我们只要说明对于每个 \(r\),在 \(n=n_{\max}\)(本题中 \(n_{\max}=10^9\))时,\(a,b,c\) 三维都不会太大即可。
事实上,我们只需将 \(r=1,n=10^9\),\(r=2,n=10^9\),……,\(r=10^4,n=10^9\) 都代入我们上面所说的做法中运行一遍,得出结果:\(a\) 始终在 \([1,790]\) 间,\(b\) 始终在 \([-199,78605]\) 间,\(c\) 始终在 \([1,39788]\) 间,因此用 long long 存储是完全够用的。
上述做法在 \(n,r\) 上界更大时并不一定能说明值域不会太大,因此如果你有能在任意 \(n,r\) 的情况下都可以直接分析出 \(a,b,c\) 三者的值域范围的做法可以在评论区分享一下。
每层递归内的时间复杂度瓶颈在于算 \(\gcd(a,b,c)\),这是 \(O(\log \min(|a|,|b|,|c|))\) 的。
总时间复杂度 \(O(t\log n\log \min(|a|_{\max},|b|_{\max},|c|_{\max}))\),本题中 \(\min(|a|_{\max},|b|_{\max},|c|_{\max})=790\),空间复杂度 \(O(1)\)。
6.CF1770F Koxia and Sequence
如果 \(n\ge 2\),考虑所有合法的 \(\{a\}\) 中,如果 \(a_{n-1}\ne a_{n}\),则可以交换 \(a_{n-1},a_n\),将这样的序列两两配对,可以发现该情况对答案是没有贡献的;如果 \(a_{n-1}=a_{n}\),则对答案的贡献相当于前 \(n-2\) 个数异或起来,此时如果 \(n-2\ge 2\),我们再考虑 \(a_{n-3}\) 和 \(a_{n-2}\) 是否相等,同理可以发现不相等对答案没有贡献,相等则只用考虑前 \(n-4\) 个数异或起来的结果。如此不断考虑可以发现,如果 \(n\) 为偶数则答案为 \(0\),如果 \(n\) 为奇数则答案为所有合法的 \(\{a\}\) 中,\(a_1\) 的异或和。
接下来我们把 \(n\) 个数的或等于 \(y\) 的限制容斥掉,对每个 \(y'\subseteq y\) 都算出 \(n\) 个数的或为 \(y'\) 的子集的答案 \(f(y')\),最终答案就是 \(\bigoplus_{y'\subseteq y} f(y')\)(\(\bigoplus\) 代表异或)。
枚举 \(y'\),现在所有 \(a_i\subseteq y'\)。考虑二进制下从低往高第 \(i\) 位的答案,如果 \(y'\) 的第 \(i\) 位为 \(0\) 则答案为 \(0\),否则,\(a\) 对第 \(i\) 位有贡献当且仅当 \(a_1\) 的第 \(i\) 位为 \(1\)。那么,我们现在要统计的就是 \(a_1\subseteq y'-2^i,a_2\sim a_n\subseteq y',2^i+\sum_{j=1}^n a_j=x\) 的方案数。即 \(\sum_{a_1+\cdots+a_n=x-2^i}[a_1\subseteq y'-2^i][a_2\subseteq y']\cdots[a_n\subseteq y']\)。由于 \([a\subseteq b]=\binom{b}{a}\bmod 2\),所以上式等于 \((\sum_{a_1+\cdots+a_n=x-2^i}\binom{y'-2^i}{a_1}\binom{y'}{a_2}\cdots\binom{y'}{a_n})\bmod 2\),由范德蒙德卷积该式等于 \(\binom{ny'-2^i}{x-2^i}\bmod 2=[x-2^i\subseteq ny'-2^i]\)。故枚举 \(y'\) 和 \(i\) 之后可以 \(O(1)\) 判断。
时间复杂度 \(O(y\log y)\),空间复杂度 \(O(1)\)。
7.LOJ#6440. 万能欧几里得
为了避免重名,将原题中的 \(P,Q,R,L\) 在下文中均用小写字母表示。
万能欧几里得算法的思想是这样的:把问题抽象成,在坐标系上有一条直线 \(y = \frac{px+r}{q}\),我们在坐标系上画出所有 \(x = k, y = k, k \in Z\) 的直线,现在有两种可合并的操作 \(U, R\),我们从左往右扫这条直线,遇到横线就执行 \(U\) 操作,遇到竖线就执行 \(R\) 操作,两者同时遇到先执行 \(U\) 操作再执行 \(R\) 操作,求做完这个操作序列之后得到的答案。直线的定义域是 \((0,l]\)。
比如说这题,我们可以维护一个三元组 \((a,b,ans)\),表示一段由 \(U,R\) 组成的序列执行完后,\(A^x=a\),\(B^{\lfloor\frac{px+r}{q}\rfloor}=b\),答案矩阵为 \(ans\)。现在假设有两个操作序列分别对应三元组 \((a_1,b_1,ans_1)\) 和 \((a_2,b_2,ans_2)\),则把两个操作序列拼起来后得到的三元组为 \((a_1a_2,b_1b_2,ans_1+a_1ans_2b_1)\)。令 \(I\) 为单位矩阵,\(O\) 为全 \(0\) 矩阵,于是 \(U\) 操作对应的三元组为 \((I,B,O)\),\(R\) 操作对应的三元组为 \((A,I,A)\)。令 \(f(p,q,r,l,U,R)\) 为上述问题的答案,为了方便我们要时刻保持直线在 \(x=0\) 处的值在 \([0,1)\) 间,即 \(0\le r\lt q\)。同时还有一个性质,\(f(p,q,r,l,U,R)\) 对应的操作序列一定以 \(R\) 结尾。
由于定义域的左端点是开区间,所以我们要先特殊处理一下一开始 \(r\ge q\) 的情况,此时相当于先执行了 \(\lfloor\frac{r}{q}\rfloor\) 次 \(R\) 操作。
如果 \(l=0\),答案显然为 \((I,I,O)\)。
如果 \(p\ge q\),则每个 \(R\) 之前至少有 \(\lfloor\frac{p}{q}\rfloor\) 个 \(U\),把每个 \(R\) 和它前面的 \(\lfloor\frac{p}{q}\rfloor\) 个 \(U\) 捆绑在一起,即 \(f(p\bmod q,q,r,l,U,U^{\lfloor\frac{p}{q}\rfloor}R)\)。
如果 \(p<q\),若此时 \(U\) 的数量为 \(0\) 则把所有 \(R\) 用快速幂一起处理,否则考虑翻转坐标系。先考虑一个简单的问题:第 \(a\) 个 \(R\) 前有几个 \(U\)?由于直线在 \(x=0\) 处的值在 \([0,1)\) 间,所以答案就是 \(R\) 的横坐标,即 \(\lfloor\frac{pa+r}{q}\rfloor\)。再考虑一个问题:第 \(a\) 个 \(U\) 前有几个 \(R\)?设有 \(b\) 个,则一定满足 \(\lfloor\frac{p(b-1)+r}{q}\rfloor\le a-1\),\(\lfloor\frac{pb+r}{q}\rfloor\ge a\),解得 \(\lfloor\frac{qa-r-1}{p}\rfloor\le b\lt \lfloor\frac{qa-r-1}{p}\rfloor+1\),因此 \(b=\lfloor\frac{qa-r-1}{p}\rfloor\)。那么我们新的问题大致就形如,在坐标系上有一条直线 \(y=\frac{qx-r-1}{p}\),从左往右扫这条直线,遇到横线就执行 \(R\) 操作,遇到竖线就执行 \(U\) 操作,两者同时遇到先执行 \(R\) 操作再执行 \(U\) 操作,求做完这个操作序列之后得到的答案。
再考虑一些细节上的问题,一个问题是我们要保证新直线的 \(r,q\) 满足 \(0\le r\lt q\),另一个问题是新的操作序列要以 \(U\) 结尾。对于前者,我们先把第一个 \(U\) 前的所有 \(R\) 和第一个 \(U\) 单独处理,把整条直线向左平移一格,新直线的解析式就变成了 \(y=\frac{qx+q-r-1}{p}\)。由于定义域的左端点是开区间,所以将新直线的 \(r\) 增大或减小 \(p\) 的倍数都不影响得到的操作序列,所以可以把 \(q-r-1\) 对 \(p\) 取模,完整写出来就是 \(f(q,p,(q-r-1)\bmod p,\lfloor\frac{pl+r}{q}\rfloor-1,R,U)\)。对于后者,我们将原操作序列中最后一个 \(U\) 后的 \(R\) 一起处理即可。
如果认为 \(P,Q,R,L\) 同阶,则时间复杂度为 \(O(n^3\log^2 P)\),空间复杂度 \(O(n^2)\)。
8.CF1091G New Year and the Factorisation Collaboration
交互库提供的内容中只有 sqrt 比较难自己计算。
我们考虑随机一个 \(x\),满足 \(x\) 与 \(n\) 互质,并询问 sqrt \(x^2\bmod n\)。假设返回了 \(x'\),那么我们知道 \(x^2\equiv x'^2\pmod n\),即 \((x-x')(x+x')\equiv 0\pmod n\)。
对于 \(n\) 的每个质因子 \(p\),也都有 \((x-x')(x+x')\equiv 0\pmod p\)。由于 \(x\not\equiv 0\pmod p\),因此 \(\gcd(x-x',x+x')=\gcd(2x,x+x')\) 是 \(2x\) 的因数进而不是 \(p\) 的倍数,所以 \(x-x'\equiv 0\pmod p\) 和 \(x+x'\equiv 0\pmod p\) 两者仅有一个成立。由于 \(x\) 是随机的,所以两者成立的概率是相等的,都为 \(\frac{1}{2}\)。
于是可以把 \(n\) 划分成两部分,\(\gcd(n,x-x')\) 和 \(\gcd(n,x+x')\),每个质因子只在恰好一者中出现且出现概率均为 \(\frac{1}{2}\)。可以看出,随机足够多次数后,就可以通过最终的划分确定 \(n\) 的质因子了。
9.CF1270I Xor on Figures
设一次操作 \((X,Y,W)\) 表示把 \(1\le i\le t\),\(a_{(x_i+X)\bmod 2^k,(y_i+Y)\bmod 2^k}\) 的位置都异或上 \(W\)。我们可以把 \(X,Y\) 完全相同的操作先合并起来,于是每个位置最多被操作一次。令 \(W_{i,j}\) 表示 \(X=i,Y=j\) 的操作中 \(W\) 的值(没操作过视为 \(0\)),我们希望矩阵 \(W\) 中非 \(0\) 的位置尽量少。
定义矩阵 \(F\),满足 \(F\) 中有且仅有 \(F_{x_i,y_i}=1\)。重新定义两个 \(2^k\times 2^k\) 的矩阵 \(A,B\) 的乘法,满足 \((A\times B)_{x,y}=\bigoplus_{i=0}^{2^k-1}\bigoplus_{j=0}^{2^k-1}A_{i,j}B_{(x-i)\bmod 2^k,(y-j)\bmod 2^k}\),其中 \(\bigoplus\) 表示异或。于是有,\(FW=A\)(\(A\) 是题中给定的矩阵)。
下面说明,可能的 \(W\) 有且仅有一种。
首先要说明,\(F^{2^k}=I\),其中 \(I_{i,j}=[i=0\vee j=0]\)。考虑 \(F^2\),发现只有 \(F^2_{2x_i\bmod 2^k,2y_i\bmod 2^k}\) 这些位置可能有值,而形如 \(F^2_{x_i+x_j\bmod 2^k,y_i+y_j\bmod 2^k}\) 的位置会被 \(F_{x_i,y_i}F_{x_j,y_j}\) 和 \(F_{x_j,y_j}F_{x_i,y_i}\) 异或两次抵消掉了。归纳可得,在 \(F^{2^k}\) 中,只有 \(F^{2^k}_{2^kx_i\bmod 2^k,2^ky_i\bmod 2^k}\) 这些位置可能有值,即只有 \(F^{2^k}_{0,0}\) 可能有值。由于 \(F\) 中原先有奇数个 \(1\) 且 \(F\) 是 \(01\) 矩阵,所以 \(F_{0,0}=1\)。
因此 \(FW=A\) 等价于 \(F^{2^{k}-1}FW=F^{2^{k}-1}A\),即 \(W=F^{2^{k}-1}A\),故 \(W\) 存在且唯一。
关于如何快速求解 \(W\),注意到 \(W=\prod_{i=0}^{k-1}F^{2^i}A\),而任意 \(F^{2^i}\) 中都至多有 \(t\) 个位置不为 \(0\),因此单次矩阵乘法的复杂度为 \(O(t2^k)\),总共要做 \(O(k)\) 次矩阵乘法。
时间复杂度 \(O(kt2^k)\),空间复杂度 \(O(2^k)\)。
10.SP20174 DIVCNT3 - Counting Divisors (cube)
前置知识:Powerful Number 筛。
下文对于函数名为小写字母的函数,用函数名为对应的大写字母的函数表示其前缀和。
令 \(d(i)=\sigma_0(i)\),\(f(i)=d(i^3)\),那么我们要求的就是 \(F(n)\)。
尝试构造积性函数 \(g\) 使得对于任意质数 \(p\) 都有 \(g(p)=f(p)\)。注意到 \(f(p)=4,d(1)=1,d(p)=2\),所以 \(g=d*d\) 是一个合法的构造(此时 \(g(p)=d(1)d(p)+d(p)d(1)=4\))。
现在要再构造一个积性函数 \(h\) 使得 \(f=h*g\),并且我们只需要预处理出 \(h\) 在质数的幂处的取值,这可以由 \(f,g\) 在质数的幂处的取值推来。注意到 \(h(p^k)\) 只和 \(k\) 有关,而和 \(p\) 无关,所以我们只用求出 \(O(\log n)\) 项 \(h\),时间复杂度 \(O(\log^2 n)\),可忽略不计。
现在我们用 dfs \(O(\sqrt n)\) 枚举所有 Powerful Number 并顺带计算出它的 \(h\),对于 Powerful Number \(x\),我们只需算出 \(G(\lfloor\frac{n}{x}\rfloor)\)。
\(G(n)=\sum_{i=1}^{n}g(i)=\sum_{i=1}^{n}\sum_{j|i} d(j)d(\frac{i}{j})=\sum_{j=1}^{n}d(j)\sum_{i=1}^{\lfloor\frac{n}{j}\rfloor} d(i)=\sum_{j=1}^{n}d(j)D(\lfloor\frac{n}{j}\rfloor)\)。
\(D(n)=\sum_{i=1}^{n}d(i)=\sum_{i=1}^{n}\sum_{j|i}1=\sum_{j=1}^n\sum_{i=1}^{\lfloor\frac{n}{j}\rfloor}1=\sum_{j=1}^n\lfloor\frac{n}{j}\rfloor\)。
为了快速计算 \(G,D\),我们可以使用数论分块,并且此时会用到的 \(G,D\) 的值一定形如 \(G(\lfloor\frac{n}{x}\rfloor),H(\lfloor\frac{n}{x}\rfloor)\) 的形式。用线性筛预处理出前 \(O(n^{\frac{2}{3}})\) 的 \(G,D\) 的值后再记忆化搜索即可达到 \(O(n^{\frac{2}{3}})\) 的复杂度。
总时间复杂度 \(O(n^\frac{2}{3})\),空间复杂度 \(O(\sqrt n)\)。
11.CF1656H Equal LCM Subsets
如果 \(A(B)\) 中存在一个数,它的某个质因子指数比 \(B(A)\) 中所有数该质因子指数都大,那这个数就一定要删掉。
如果不存在这样的数了,要么是有集合被删空了,此时无解;否则,此时两集合的 \(\text{lcm}\) 相同。
为了快速找到 \(A\) 中是否存在这样的数,我们要找到是否有 \(a_i\in A\),不满足 \(a_i|\text{lcm}_{b_j\in B}\{b_j\}\)(\(B\) 中是对称的)。而 \(a_i|\text{lcm}_{b_j\in B}\{b_j\}\Leftrightarrow \text{lcm}_{b_j\in B}\{\gcd(b_j,a_i)\}=a_i\Leftrightarrow \gcd_{b_j\in B}\{\frac{a_i}{\gcd(b_j,a_i)}\}=1\)(因为三者都等价于对于任意质数 \(p\),\(\max_{b_j\in B}\{v_p(b_j)\}\ge v_p(a_i)\))。因此可以开 \(n\) 棵线段树来维护所有 $\frac{a_i}{\gcd(b_j,a_i)}$ 的 \(\gcd\),删掉 \(b_j\) 就相当于在 \(n\) 棵线段树上单点修改 \(b_j\) 为 \(1\)。
如果认为 \(n,m\) 同阶,\(a,b\) 同阶,则时间复杂度 \(O(n^2(\log n+\log a))\),空间复杂度 \(O(n^2)\)。
12.CF1566H Xor-quiz
首先我们只用问 \([1,c]\) 中 \(\mu\ne 0\) 的数,在题目范围内这些数的数量级是不超过 \(\lceil0.65c\rceil\) 的,也就是说我们可以获得全部信息。
现在对任意满足其所有质因子出现次数都为 \(1\) 的 \(k\),都知道了与 \(k\) 互质的数的异或和,通过反演可以求出所有不同质因子的乘积恰为 \(k\) 的数的异或和。
也就是说现在有若干集合,要求每个集合内选一些数使得它们的异或和等于某个给定值,且所有集合内选的数的个数总和为 \(n\)。建出每个集合内的线性基,因为有解所以每个集合的给定值一定可以被该集合的线性基表出。对于一个大小为 \(x\) 的集合,假设建出了大小为 \(y\) 的线性基,则有 \(x-y\) 个集合内的数不在线性基中(有 \(x-y\) 个自由元)。所以为了异或出给定的数我们要现在自由元中选出一个子集,并将该子集通过线性基内的数异或成给定的数,可以发现恰有 \(2^{x-y}\) 种方案。
打表可以发现,在 \(c=10^6\) 时最大的集合为 \(210\) 对应的集合,其中有 \(260\) 个数,所以在自由元中枚举每个子集是不现实的,由于数据随机,我们考虑在自由元中随机一些子集。还可以发现,没有自由元的集合只有唯一一种选法,因此可以忽略,打表可知有自由元的集合的大小总和在 \(c=10^6\) 时为 \(68873\),这说明了每个集合内能选出的最大合法子集的大小减最小合法子集的大小不会太大。那么在随完子集后用 bitset 优化 dp 即可。