模板题求助qaq
查看原帖
模板题求助qaq
320815
STL_Lover楼主2020/7/30 20:41

调了一晚上了,跟题解对后改了一下反而死循环了(

#include<cstdio>
#include<cstring>
using namespace std;

int n,m,mod,tmp,x,y,k;
const int maxn=100010;
int a[maxn],b1[maxn<<2],b2[maxn<<2],d[maxn<<2],ls[maxn<<2],rs[maxn<<2];
void pushup(int p)
{
    d[p]=(d[p<<1]+d[p<<1|1])%mod;
}
void pushdown(int p)
{
    d[p<<1]=(d[p<<1]*b1[p]+b2[p]*(rs[p<<1]-ls[p<<1]+1))%mod;
    d[p<<1|1]=(d[p<<1|1]*b1[p]+b2[p]*(rs[p<<1|1]-ls[p<<1|1]+1))%mod;
    b1[p<<1]=(b1[p<<1]*b1[p])%mod;
    b1[p<<1|1]=(b1[p<<1|1]*b1[p])%mod;
    b2[p<<1]=(b2[p<<1]*b1[p]+b2[p])%mod;
    b2[p<<1|1]=(b2[p<<1|1]*b1[p]+b2[p])%mod;
    b1[p]=1;b2[p]=0;
}
void build(int l,int r,int p)
{
    ls[p]=l;rs[p]=r;b1[p]=1;
    if(l==r){d[p]=a[l];return;}
    int mid=l+r>>1;
    if(l<=mid) build(l,mid,p<<1);
    if(mid<r) build(mid+1,r,p<<1|1);
    pushup(p);
}
void update1(int s,int t,int c,int p)
{
    if(s<=ls[p]&&rs[p]<=t)
    {
        d[p]*=c;b1[p]*=c;b2[p]*=c;
        d[p]%=mod;b1[p]%=mod;b2[p]%=mod;
        return;
    }
    pushdown(p);
    int mid=s+t>>1;
    if(s<=mid) update1(s,mid,c,p<<1);
    if(mid<t) update1(mid+1,t,c,p<<1|1);
    pushup(p);
}
void update2(int s,int t,int c,int p)
{
    if(s<=ls[p]&&rs[p]<=t)
    {
        b2[p]+=c;d[p]+=c*(rs[p]-ls[p]+1);
        b2[p]%=mod;d[p]%=mod;
        return;
    }
    pushdown(p);
    int mid=s+t>>1;
    if(s<=mid) update2(s,mid,c,p<<1);
    if(mid<t) update2(mid+1,t,c,p<<1|1);
    pushup(p);
}
int query(int s,int t,int p)
{
    if(s<=ls[p]&&rs[p]<=t) return d[p];
    pushdown(p);
    int mid=s+t>>1,sum=0;
    if(s<=mid) {sum+=query(s,mid,p<<1);sum%=mod;}
    if(mid<t) {sum+=query(mid+1,t,p<<1|1);sum%=mod;}
    pushup(p);
    return sum;
}
int main()
{
    scanf("%d%d%d",&n,&m,&mod);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(1,n,1);
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&tmp);
        if(tmp==1)
        {
            scanf("%d%d%d",&x,&y,&k);
            update1(x,y,k,1);
        }
        else if(tmp==2)
        {
            scanf("%d%d%d",&x,&y,&k);
            update2(x,y,k,1);
        }
        else
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",query(x,y,1));
        }
    }
    return 0;
}
2020/7/30 20:41
加载中...