BZOJ 4105: [Thu Summer Camp 2015]平方运算

来源:互联网 发布:高能金域名都物业 编辑:程序博客网 时间:2024/06/06 07:03

这道题好劲啊 乍一看就知道是线段树 然而不知道怎么做
于是又看题解 愉快地引用结论 (数据里的p比较特殊???

首先平方之后取模 这种东西显然是有环的 然后(根据结论…..)其实最大环只有60。。
所以我们只需要预处理出每个数是否在环上 以及他的next是什么
接着每次平方操作就相当于在环上走一步 (当然你要算出环上每个位置的和)
不在环上的呢? 暴力走。。同样的 他到环的距离不会很长 (有人说<11)

故时间复杂度为 O(60nlogn)
注意 我说的环不只是对单个数自己平方形成的 而是对一个数列一起平方
实在不懂的话就看看代码里的upd和pushdown吧 不难懂。。

#include<bits/stdc++.h>using namespace std;const int N=100010,M=10000;inline int read(){    char ch=getchar(); int x=0,f=1;    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0'; ch=getchar();}    return x*f;}inline int lcm(int a,int b){return a/__gcd(a,b)*b;}int p,s[N<<2],w[N<<2],a[N],len[N<<2],cir[N<<2][62];bool inc[M],in[N<<2]; int nx[M],lz[N<<2];void init(int x,int v){    s[x]=v; in[x]=inc[v];    if(in[x]){        len[x]=1,cir[x][w[x]=0]=v;        for(int u=nx[v];u!=v;u=nx[u]) cir[x][len[x]++]=u;    }}void upd(int x){    int lc=x<<1,rc=lc|1;    s[x]=s[lc]+s[rc],in[x]=in[lc]&in[rc];    if(in[x]){        len[x]=lcm(len[lc],len[rc]);        int tx=w[lc],ty=w[rc];        for(int i=0;i<len[x];++i){            cir[x][i]=cir[lc][tx++]+cir[rc][ty++];            if(tx==len[lc])tx=0; if(ty==len[rc])ty=0;        }        w[x]=0;    }}void bulid(int x,int l,int r){    if(l==r){        init(x,a[l]); return;    }    int lc=x<<1,rc=lc|1,mid=(l+r)>>1;    bulid(lc,l,mid),bulid(rc,mid+1,r);    upd(x);}void pushdown(int x){    int c=lz[x],lc=x<<1,rc=lc|1; lz[x]=0;    lz[lc]+=c,lz[rc]+=c; w[lc]+=c,w[rc]+=c;    if(w[lc]>=len[lc])w[lc]%=len[lc]; s[lc]=cir[lc][w[lc]];    if(w[rc]>=len[rc])w[rc]%=len[rc]; s[rc]=cir[rc][w[rc]];}void change(int x,int l,int r,int ql,int qr){    if(l==ql&&r==qr&&in[x]){        ++lz[x],++w[x]; if(w[x]==len[x])w[x]=0;        s[x]=cir[x][w[x]]; return;    }    if(l==r){init(x,nx[s[x]]); return;}    if(lz[x])pushdown(x); int mid=(l+r)>>1,lc=x<<1,rc=lc|1;    if(qr<=mid)change(lc,l,mid,ql,qr);    else if(ql>mid)change(rc,mid+1,r,ql,qr);    else change(lc,l,mid,ql,mid),change(rc,mid+1,r,mid+1,qr);    upd(x);}int query(int x,int l,int r,int ql,int qr){    if(l==ql&&r==qr)return s[x];    if(lz[x])pushdown(x); int mid=(l+r)>>1,lc=x<<1,rc=lc|1;    if(qr<=mid)return query(lc,l,mid,ql,qr);    else if(ql>mid)return query(rc,mid+1,r,ql,qr);    return query(lc,l,mid,ql,mid)+query(rc,mid+1,r,mid+1,qr);}bool v[M];int main(){    int n=read(),m=read(),i; p=read();    for(i=1;i<=n;++i)a[i]=read();    for(i=0;i<p;++i)nx[i]=(i*i)%p,inc[i]=1;    for(i=0;i<p;++i) if(!v[i]){        int y=i; while(!v[y])v[y]=1,y=nx[y];        for(int t=i;t!=y;t=nx[t])inc[t]=0;    }    bulid(1,1,n);    while(m--){        int u=read(),l=read(),r=read();        if(!u) change(1,1,n,l,r);        else printf("%d\n",query(1,1,n,l,r));    }    return 0;}
1 0
原创粉丝点击