2018.10.18 bzoj4105: [Thu Summer Camp 2015]平方运算（线段树）

版权声明：随意转载哦......但还是请注明出处吧： https://blog.csdn.net/dreaming__ldx/article/details/83118569

传送门
线段树妙题。

---

显然平方几次就会循环（打表证明不解释）。
然后所有环长度的 $lcm$不大于70.
因此维护一下当前区间中的节点是否全部在环上。
不是直接暴力到叶子节点修改。
否则整体打标记。
简单合并一下就行了。（细节见代码）
代码：

#include<bits/stdc++.h>
#define N 100005
#define lc (p<<1)
#define rc (p<<1|1)
#define mid (T[p].l+T[p].r>>1)
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
int n,m,mod,a[N],tot=0,tim[N],cnt=0,nxt[10005];
bool inloop[10005],vis[10005];
struct Node{int l,r,sum,cir[70],pos,tag,len;bool in;}T[N<<2];
inline void solve(int p,int val){
	if(!(T[p].in=inloop[T[p].sum=val]))return;
	T[p].len=1,T[p].cir[T[p].pos=0]=val;
	for(int i=nxt[val];i!=val;i=nxt[i])T[p].cir[T[p].len++]=i;
}
inline int lcm(int a,int b){return a/__gcd(a,b)*b;}
inline void pushup(int p){
	T[p].sum=T[lc].sum+T[rc].sum;
	T[p].in=T[lc].in&T[rc].in;
	if(!T[p].in)return;
	T[p].len=lcm(T[lc].len,T[rc].len);
	int tl=T[lc].pos,tr=T[rc].pos; 
	for(int i=0;i<T[p].len;++i){
		T[p].cir[i]=T[lc].cir[tl++]+T[rc].cir[tr++];
		if(tl==T[lc].len)tl=0;
		if(tr==T[rc].len)tr=0;
	}
	T[p].pos=0;
}
inline void pushnow(int p,int v){
	T[p].tag+=v,T[p].pos+=v;
	if(T[p].pos>=T[p].len)T[p].pos%=T[p].len;
	T[p].sum=T[p].cir[T[p].pos];
}
inline void pushdown(int p){if(T[p].tag)pushnow(lc,T[p].tag),pushnow(rc,T[p].tag),T[p].tag=0;}
inline void build(int p,int l,int r){
	T[p].l=l,T[p].r=r;
	if(l==r)return solve(p,a[l]);
	build(lc,l,mid),build(rc,mid+1,r),pushup(p);
}
inline void update(int p,int ql,int qr){
	if(ql<=T[p].l&&T[p].r<=qr&&T[p].in)return pushnow(p,1);
	if(T[p].l==T[p].r)return solve(p,nxt[T[p].sum]);
	pushdown(p);
	if(qr<=mid)update(lc,ql,qr);
	else if(ql>mid)update(rc,ql,qr);
	else update(lc,ql,mid),update(rc,mid+1,qr);
	pushup(p);
}
inline int query(int p,int ql,int qr){
	if(ql<=T[p].l&&T[p].r<=qr)return T[p].sum;
	pushdown(p);
	if(qr<=mid)return query(lc,ql,qr);
	if(ql>mid)return query(rc,ql,qr);
	return query(lc,ql,mid)+query(rc,mid+1,qr);
}
int main(){
	n=read(),m=read(),mod=read();
	for(int i=1;i<=n;++i)a[i]=read();
	for(int i=0;i<mod;++i)nxt[i]=i*i%mod,inloop[i]=true;
	for(int i=0;i<mod;++i)
		if(!vis[i]){
			int tmp=i;
			while(!vis[tmp])vis[tmp]=1,tmp=nxt[tmp];
			for(int j=i;j!=tmp;j=nxt[j])inloop[j]=false;
		}
	build(1,1,n);
	while(m--){
		int op=read(),l=read(),r=read();
		if(!op)update(1,l,r);
		else printf("%d\n",query(1,l,r));
	}
	return 0;
}