SHOI2017 相逢是问候(扩展欧拉定理+线段树)

题意

https://loj.ac/problem/2142

思路

一个数如果要作为指数,那么它不能直接对模数取模,这是常识;
诸如 c c c c . . c^{c^{c^{c..}}} cccc.. 的函数递增飞快,不是高精度可以描述的,这也是常识。
所以,此题要用到很多数论知识。

欧拉函数

定义

φ ( n ) \varphi(n) φ(n) [ 1 , n ] [1,n] [1,n] 中与 n n n 互质的正整数个数(包括 1 1 1)。

通式

φ ( n ) = n ∏ p ∣ n ( 1 − 1 p ) \displaystyle \varphi(n)=n\prod_{p|n}(1-{1\over p}) φ(n)=npn(1p1) 其中 p p p x x x 的质因子。
如何理解这个式子呢?可以粗略这样理解: ( 1 − 1 p ) \displaystyle (1-{1\over p}) (1p1) 意思就是筛掉所有能被 p p p 整除的数,当然这种理解方法是错误的,只能算是感性理解,方便记忆。

前几项

摘自维基百科:
SHOI2017 相逢是问候(扩展欧拉定理+线段树)_第1张图片

性质

几条比较重要的性质:
gcd ⁡ ( m , n ) = 1 \gcd(m,n)=1 gcd(m,n)=1,有 φ ( m n ) = φ ( m ) φ ( n ) \varphi(mn)=\varphi(m)\varphi(n) φ(mn)=φ(m)φ(n)
m = 2 m=2 m=2,不难得出当 n n n 为奇数时, φ ( 2 n ) = φ ( n ) \varphi(2n)=\varphi(n) φ(2n)=φ(n)
另外,对于质数 p p p,有 φ ( p ) = p − 1 \varphi(p)=p-1 φ(p)=p1

欧拉定理

a φ ( n ) ≡ 1 ( m o d n ) a^{\varphi(n)} \equiv 1 \pmod n aφ(n)1(modn) 其中 gcd ⁡ ( a , n ) = 1 \gcd(a,n)=1 gcd(a,n)=1
n n n 为质数的情况就是著名的费马小定理

扩展欧拉定理

a b = { a b    b < φ ( p ) a b m o d    φ ( p ) + φ ( p ) b ≥ φ ( p ) ( m o d p ) a^b=\begin{cases} a^b\quad \quad \quad \quad\quad \quad \ \ b<\varphi(p)\\ a^{b\mod\varphi(p)+\varphi(p)}\quad b\geq \varphi(p) \end{cases}\pmod p ab={ab  b<φ(p)abmodφ(p)+φ(p)bφ(p)(modp)
扩展欧拉定理把幂的层数和模数都降了下来,用它就可以解决本题了。

回归本题

先考虑暴力做法再尝试优化。我们可以给每个数一个标记,表示多少层 c c c 上才是 a i a_i ai。然后每次只需计算这个 c c c . . . a i c^{c^{c^{...^{a_i}}}} ccc...ai 即可。
套用扩展欧拉定理,假设标记为 3 3 3,就是计算 c c c a i m o d    p c^{c^{c^{a_i}}} \mod p cccaimodp 的值。
分两类讨论,当 c c a i < φ ( p ) c^{c^{a_i}}<\varphi(p) ccai<φ(p),问题就转化成了求 c c a i m o d    φ ( p ) c^{c^{a_i}}\mod \varphi(p) ccaimodφ(p) 的值;否则,只需在此基础上加 φ ( p ) \varphi(p) φ(p) 即可,而判断大小只需在快速幂上传一个标记就行了。
一路递归下去,当模数变成 1 1 1 时直接返回 0 0 0 即可。
考虑优化。由于模数给定,那么变化显然是从 p p p φ ( p ) \varphi(p) φ(p) 再到 φ ( φ ( p ) ) \varphi(\varphi(p)) φ(φ(p)) 这样一路变化的。变化不会超过 log ⁡ p \log p logp 层,那么总标记次数就不超过 n log ⁡ p n\log p nlogp 次。
可以用线段树维护一个区间,那么再修改一个区间时,也维护一个区间的最小修改次数的最大值,当它对应的模数已经是 1 1 1 时,便不进行修改。再修改时,要算出 c c c . . . a i c^{c^{c^{...^{a_i}}}} ccc...ai 的值,最多修改 n log ⁡ n n\log n nlogn 次,最深又有 log ⁡ \log log 层,算快速幂又有一个 log ⁡ \log log。三个 log ⁡ \log log 是过不了本题的。
然而快速幂可以直接分块预处理,由于模数不超过 log ⁡ n \log n logn 个,则分别预处理
c i m o d    p k , c 50000 i m o d    p k c^{i} \mod p_k,c^{50000i} \mod p_k cimodpk,c50000imodpk 的值,以及和 p k p_k pk 的大小关系即可。复杂度降至两个 log ⁡ \log log
查询直接区间求和,一个 log ⁡ \log log

代码

#include
#define FOR(i,x,y) for(int i=(x),i##END=(y);i<=i##END;++i)
#define DOR(i,x,y) for(int i=(x),i##END=(y);i>=i##END;--i)
typedef long long LL;
using namespace std;
const int N=5e4+5;
const int _N=5e4;
int P[105],a[N],tot,n,m,c;
LL PW[2][N][105];bool FL[2][N][105];
//PW[i][j][k]  c^(i*_N+j)%P[k]
//FL[i][j][k] [c^(i*_N+j)>=P[k]]
LL phi(LL n)
{
	LL res=n;
	for(LL i=2;i*i<=n;i++)if(n%i==0)
	{
		res=res/i*(i-1);
		while(n%i==0)n/=i;
	}
	if(n>1)res=res/n*(n-1);
	return res;
}
struct node
{
	int Mi,sum;
	node operator +(const node &_)const
	{
		return (node){min(Mi,_.Mi),(sum+_.sum)%P[0]};
	}
}nd[N<<2];
void build(int k,int l,int r,int *arr)
{
	if(l==r)
	{
		nd[k].Mi=0;
		nd[k].sum=arr[l];
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid,arr);
	build(k<<1|1,mid+1,r,arr);
	nd[k]=nd[k<<1]+nd[k<<1|1];
}
LL Pow(LL p,int id,bool &flag)//c^p%P[id] flag=[c^p>=P[id]]
{
	flag=FL[0][p%_N][id]||FL[1][p/_N][id]||(PW[0][p%_N][id]*PW[1][p/_N][id]>=P[id]);
	return PW[0][p%_N][id]*PW[1][p/_N][id]%P[id];
}
LL calc(int x,LL y)
{
	if(y>=P[x])y=y%P[x]+P[x];
	DOR(i,x,1)
	{
		bool flag;
		y=Pow(y,i-1,flag);
		if(flag)y+=P[i-1];
	}
	return y%P[0];
}
void update(int k,int L,int R,int l,int r)
{
	if(L<=l&&r<=R&&nd[k].Mi>=tot)return;
	if(l==r)
	{
		nd[k].sum=calc(++nd[k].Mi,a[l]);
		return;
	}
	int mid=(l+r)>>1;
	if(L<=mid)update(k<<1,L,R,l,mid);
	if(R>mid)update(k<<1|1,L,R,mid+1,r);
	nd[k]=nd[k<<1]+nd[k<<1|1];
}
int query(int k,int L,int R,int l,int r)
{
	if(L<=l&&r<=R)return nd[k].sum;
	int mid=(l+r)>>1;
	if(R<=mid)return query(k<<1,L,R,l,mid);
	else if(L>mid)return query(k<<1|1,L,R,mid+1,r);
	else return (query(k<<1,L,R,l,mid)+query(k<<1|1,L,R,mid+1,r))%P[0];
}
void pre_compute()
{
	FOR(i,0,tot)
	{
		PW[0][0][i]=1,FL[0][0][i]=(1>=P[i]);
		FOR(j,1,_N)
		{
			PW[0][j][i]=PW[0][j-1][i]*c;
			FL[0][j][i]=FL[0][j-1][i]|(PW[0][j][i]>=P[i]);
			if(PW[0][j][i]>=P[i])PW[0][j][i]%=P[i];
		}
		
		PW[1][0][i]=1,FL[1][0][i]=(1>=P[i]);
		FOR(j,1,_N)
		{
			PW[1][j][i]=PW[1][j-1][i]*PW[0][_N][i];
			FL[1][j][i]=FL[1][j-1][i]|(PW[1][j][i]>=P[i]);
			if(PW[1][j][i]>=P[i])PW[1][j][i]%=P[i];
		}
	}
}

int main()
{
	scanf("%d%d%d%d",&n,&m,&P[tot=0],&c);
	FOR(i,1,n)scanf("%d",&a[i]);
	while(P[tot]!=1)
	{
		P[tot+1]=phi(P[tot]);
		tot++;
	}
	P[++tot]=1;
	pre_compute();
	build(1,1,n,a);
	while(m--)
	{
		int op,l,r;
		scanf("%d%d%d",&op,&l,&r);
		if(op==0)update(1,l,r,1,n);
		else printf("%d\n",query(1,l,r,1,n));
	}
    return 0;
}

你可能感兴趣的:(题目)