[BZOJ4869][洛谷P3747][六省联考2017]相逢是问候

Solution

• 建议先做这题
• 根据扩展欧拉定理$$a^b\%p\equiv a^{b\%\varphi (p)+\varphi (p)}\%p_{，}b\ge \varphi (p)$$$$a^b\%p\equiv a^{b\%\varphi (p)}\%p，b<\varphi (p)$$
• 这题核心代码如下

inline int solve(int p) // 递归处理模数为p的答案
{
   if (p == 1) return 0;
   return ksm(2, solve(phi[p]) + phi[p], p); // ksm(x,y) = x ^ y
}

• 发现每次递归都会将 $p$ 变为 $\varphi (p)$
• 当 $p$ 是偶数时， $p$ 变 $\varphi (p)$ 至少除以 $2$；当 $p$ 是奇数时，根据 $\varphi (p)$ 的计算公式 ($q[i]$ 为 $p$ 的所有质因子，且互不相同)$$p\ast {\Pi }_{i=1}^k(q[i]-1)÷q[i]$$
• $q[i]-1$必有偶数，那么$\varphi (p)$是偶数
• 可知 $p$ 会在 $O(\log p)$ 内次变为 $1$，结束递归

---

• 而上题中，对扩展欧拉定理的应用为：$$2^{2^{2^{2^{...}}}}\%p\equiv 2^{2^{2^{2^{...}}}\%\varphi (p)+\varphi (p)}$$
• 假设 $2$ 的个数有限，等式左边有 $x$ 个 $2$， 那么等式右边的指数，也就是递归处理的部分，有 $x-1$ 个 $2$
• 也就是说，每次递归减少了一个 $2$
• 那么减少 $O(\log p)$ 个 $2$ 后必定结束递归，即当 $2$ 的个数超过 $\log p$ 时，答案全部一样
• 回到本题，可以得出一个数的修改次数上限为 $O(\log p)$
• 那么维护线段树，如果整个区间修改次数都到上限了，就 $return$
• 易得一共会碰到叶子节点 $O(n\log p)$ 次

---

• 为了方（偷）便（懒） ，记 $b[i][j]$ 为 $c^{c^{c^{..{.}^{a_i}}}}$ (共 $j$ 个 $c$，注意没有取模），记 $f(i,j,p)$ 为 $c^{c^{c^{..{.}^{a_i}}}}\%p$ (共 $j$ 个 $c$)，那么有递推式$$f(i,j,p)=c^{f(i,j-1,\varphi (p))+(b[i][j-1]<\varphi (p)?0:\varphi (p))}$$
• 预处理几个值在 $1e8$ 内的 $b[i][j]$ 就好了，显然 $j$ 都比较小，$j$ 稍大一点就是天文数字了，肯定比 $\varphi (p)$ 大，当然，注意特判 $c=1$ 和 $j$ 较大的情况
• $p$ 有 $1e8$，不能用数组把所有的 $\varphi (x),x\le p$ 都存下来，发现用到的 $x$ 只有 $p,\varphi (p),\varphi (\varphi (p)),...,1$，考虑预处理出所有这样的 $x$，并存下它们的 $\varphi$，记这些 $x$ 分别为 $d[m..1]$（即 $d[k-1]=\varphi (d[k])$），那么上式改写为$$f(i,j,k)=c^{f(i,j-1,k-1)+(b[i][j-1]<d[k-1]?0:d[k-1])}$$
• 注意递归边界：$j=0$ 或 $k=1$
• 由于用到快速幂，上式时间复杂度 $O({\log }^{2}p)$
• 发现 $f(i,j,m)$ 不能直接由 $f(i,j-1,m)$ 得来，也就是说，每次到叶子节点都要重新计算，那么总复杂度 $O(n\log n{\log }^{2}p)$，直接爆炸，考虑优化掉快速幂

---

• 对于这$m$个模数，分别预处理 $c1[i]$ 为 $c^{i\ast 20000}$，$c2[i]$ 为 $c^i$，那么 $$c^y=c1[y/20000]\ast c2[y\%20000]$$
• 然后就可以 $O(1)$ 快速幂了
• 总时间复杂度 $O(n(\log n\log p+{\log }^{2}p))$

Code

#include <bits/stdc++.h>

using namespace std;

#define ll long long
#define p2 p << 1
#define p3 p << 1 | 1

template <class t>
inline void read(t & res)
{
	char ch;
	while (ch = getchar(), !isdigit(ch));
	res = ch ^ 48;
	while (ch = getchar(), isdigit(ch))
	res = res * 10 + (ch ^ 48);
}

const int e = 5e4 + 5, o = 5e4 + 5, bl = 2e4, z = 105;
int a[e], cnt[o * 4], c1[z][20005], c2[z][20005], n, m, c, mod, phi[e];
int num[e], d[e], c4, c3[e], b[e][20], sum[o * 4];

inline void upt(int &x, int y)
{
	x = y;
	if (x >= mod) x -= mod;
}

inline int solve(int x)
{
	int s = sqrt(x), res = x, i;
	for (i = 2; i <= s; i++)
	if (x % i == 0)
	{
		while (x % i == 0) x /= i;
		res /= i;
		res *= i - 1;
	}
	if (x > 1)
	{
		res /= x;
		res *= x - 1;
	}
	return res;
}

inline void init()
{	
	d[d[0] = 1] = mod;
	for (;;)
	{
		d[0]++;
		d[d[0]] = solve(d[d[0] - 1]);
		if (d[d[0]] == 1) break;
	}
	reverse(d + 1, d + d[0] + 1);
	int i, j;
	ll y = 1;
	c3[0] = 1;
	for (i = 1; i < bl; i++) 
	{
		if (y <= 1e8)
		{
			y = y * c; 
			c3[i] = y;
			if (y <= 1e8) c4 = i;
		}
		else break;
	}
	for (i = 1; i <= d[0]; i++)
	{
		c1[i][0] = c2[i][0] = 1;
		for (j = 1; j < bl; j++) c2[i][j] = (ll)c2[i][j - 1] * c % d[i];
		c1[i][1] = (ll)c2[i][bl - 1] * c % d[i];
		for (j = 2; j < bl; j++) c1[i][j] = (ll)c1[i][j - 1] * c1[i][1] % d[i];
	}
	for (i = 1; i <= n; i++)
	{
		b[i][0] = a[i];
		for (j = 1; j <= 6; j++)
		{
			int x = b[i][j - 1], p1 = x / bl, p0 = x % bl;
			if (c3[p0] > 1e8 || p1 || p0 > c4) break;
			b[i][j] = c3[p0];
			num[i] = j;
		}
	}
}

inline int ksm(int y, int id)
{
	return (ll)c1[id][y / bl] * c2[id][y % bl] % d[id]; 
}

inline int f(int i, int j, int mod)
{
	if (mod == 1) return 0;
	if (j == 0) return a[i] % d[mod];
	if (c == 1 || (c != 1 && j - 1 <= num[i] && b[i][j - 1] < d[mod - 1])) 
	return ksm(f(i, j - 1, mod - 1), mod);
	else return ksm(f(i, j - 1, mod - 1) + d[mod - 1], mod);
}

inline void collect(int p)
{
	cnt[p] = min(cnt[p2], cnt[p3]);
	upt(sum[p], sum[p2] + sum[p3]);
}

inline void build(int l, int r, int p)
{
	if (l == r)
	{
		sum[p] = a[l];
		return;
	}
	int mid = l + r >> 1;
	build(l, mid, p2);
	build(mid + 1, r, p3);
	collect(p);
}

inline void update(int l, int r, int s, int t, int p)
{
	if (cnt[p] > d[0]) return;
	if (l == r)
	{
		cnt[p]++;
		sum[p] = f(l, cnt[p], d[0]);
		return;
	}
	int mid = l + r >> 1;
	if (s <= mid) update(l, mid, s, t, p2);
	if (t > mid) update(mid + 1, r, s, t, p3);
	collect(p);
}

inline int query(int l, int r, int s, int t, int p)
{
	if (l == s && r == t) return sum[p];
	int mid = l + r >> 1, res = 0;
	if (t <= mid) res = query(l, mid, s, t, p2);
	else if (s > mid) res = query(mid + 1, r, s, t, p3);
	else upt(res, query(l, mid, s, mid, p2) + query(mid + 1, r, mid + 1, t, p3));
	return res;
}

int main()
{
	read(n); read(m); read(mod); read(c);
	int i, opt, l, r;
	for (i = 1; i <= n; ++i) read(a[i]);
	init();
	build(1, n, 1);
	while (m--)
	{
		read(opt);
		read(l);
		read(r);
		if (!opt) update(1, n, l, r, 1);
		else printf("%d\n", query(1, n, l, r, 1));
	}
	return 0;
}