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

Solution

太菜了……我打表打出来的循环节是真正的循环节$+1$，导致没有发现规律……
发现循环节的$lcm$最大为$60$，那么线段树每个区间直接维护区间循环节就可以了。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010,Maxm=10000;
const int inf=2147483647;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int gcd(int a,int b){return((!a)?b:gcd(b%a,a));}
int lcm(int a,int b){return a/gcd(a,b)*b;}
int n,m,p,a[Maxn];
int vis[Maxm],tim=0,fir[Maxm],len[Maxm];
struct Seg{int l,r,lc,rc,len,pos,c,tag,s[62];bool flag;}tr[Maxn<<1];
int tot=0;
void Modify(int x,int v)
{
    v%=tr[x].len;
    tr[x].tag+=v;
    int nxt=tr[x].pos+v;if(nxt>tr[x].len)nxt-=tr[x].len;
    tr[x].c=tr[x].s[tr[x].pos=nxt];
}
void w(int x)
{
    tr[x].c=tr[x].c*tr[x].c%p;
    if(tr[x].c==tr[x].s[1])tr[x].flag=true,tr[x].pos=1;
}
void up(int x)
{
    int lc=tr[x].lc,rc=tr[x].rc;
    tr[x].c=tr[lc].c+tr[rc].c;
    tr[x].flag=tr[lc].flag&tr[rc].flag;
    if(tr[x].flag)
    {
        tr[x].len=lcm(tr[lc].len,tr[rc].len),tr[x].pos=1;
        int L=tr[lc].pos,R=tr[rc].pos;
        for(int i=1;i<=tr[x].len;i++)
        {
            tr[x].s[i]=tr[lc].s[L]+tr[rc].s[R];
            L=((L==tr[lc].len)?1:L+1),R=((R==tr[rc].len)?1:R+1);
        }
    }
}
void down(int x)
{
    int t=tr[x].tag;
    if(!t)return;
    Modify(tr[x].lc,t),Modify(tr[x].rc,t);tr[x].tag=0;
}
void build(int l,int r)
{
    int x=++tot;
    tr[x].l=l;tr[x].r=r;tr[x].tag=0;
    if(l==r)
    {
        int ll=0,t=a[l],tmp[62];++tim;
        tr[x].c=t;
        while(vis[t]!=tim)
        {
            vis[t]=tim;
            tmp[++ll]=t;t=t*t%p;
        }
        if(t==a[l])tr[x].flag=true,tr[x].pos=1;else tr[x].flag=false;
        for(int i=1;i<=ll;i++)
        if(tmp[i]==t)
        {
            tr[x].len=ll-i+1;
            for(int j=i;j<=ll;j++)tr[x].s[j-i+1]=tmp[j];
            break;
        }
        return;
    }
    int mid=l+r>>1;
    tr[x].lc=tot+1,build(l,mid);
    tr[x].rc=tot+1,build(mid+1,r);
    up(x);
}
void modify(int x,int l,int r)
{
    if(tr[x].l==l&&tr[x].r==r&&tr[x].flag){Modify(x,1);return;}
    if(tr[x].l==tr[x].r){w(x);return;}
    int mid=tr[x].l+tr[x].r>>1,lc=tr[x].lc,rc=tr[x].rc;
    down(x);
    if(r<=mid)modify(lc,l,r);
    else if(l>mid)modify(rc,l,r);
    else modify(lc,l,mid),modify(rc,mid+1,r);
    up(x);
}
int query(int x,int l,int r)
{
    if(tr[x].l==l&&tr[x].r==r)return tr[x].c;
    int mid=tr[x].l+tr[x].r>>1,lc=tr[x].lc,rc=tr[x].rc;
    down(x);
    if(r<=mid)return query(lc,l,r);
    if(l>mid)return query(rc,l,r);
    return query(lc,l,mid)+query(rc,mid+1,r);
}
int main()
{
    n=read(),m=read(),p=read();
    for(int i=1;i<=n;i++)a[i]=read();
    build(1,n);
    while(m--)
    {
        int op=read(),l=read(),r=read();
        if(!op)modify(1,l,r);
        else printf("%d\n",query(1,l,r));
    }
}