P3373 【模板】线段树 2

题解

样例

输入：

8 10 571373
5929 7152 8443 6028 8580 5449 8473 4237
2 4 8 4376
1 2 8 9637
2 2 6 7918
2 5 8 5681
3 2 8
1 1 5 6482
3 1 5
1 5 8 8701
2 5 8 7992
2 5 8 7806

输出

478836
562114

代码爆0  inginging

#include<bits/stdc++.h>

using namespace std;

const int maxn=1e5+10;
int n,m,x,y,v,opr,mod;
int a[maxn],add[maxn*4];
long long sum[maxn*4];
long long ans;

int qsm(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b%2==1) ans=(ans%mod*a%mod)%mod;
        a=a*a%mod;
        b/=2;
    }
    return ans;
}

inline int read()
{
    int ans=0;
    char last=‘ ‘,ch=getchar();
    while(ch<‘0‘||ch>‘9‘) last=ch,ch=getchar();
    while(ch>=‘0‘&&ch<=‘9‘) ans=ans*10+ch-‘0‘,ch=getchar();
    if(last==‘-‘) ans=-ans;
    return ans;
}

void build(int k,int l,int r)
{
    if(l==r)
    {
        sum[k]=a[l];  return ;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

void Add1(int k,int l,int r,int p)
{
    add[k]+=p;
    sum[k]*=(long long)qsm(p,(r-l+1));
}

void Add2(int k,int l,int r,int p)
{
    add[k]+=p;
    sum[k]+=(long long)(r-l+1)*p;
}

void pushdown1(int k,int l,int r,int mid)
{
    if(add[k]==0) return ;
    Add1(k<<1,l,mid,add[k]);
    Add1(k<<1|1,mid+1,r,add[k]);
    add[k]=0;
}

void pushdown2(int k,int l,int r,int mid)
{
    if(add[k]==0) return ;
    Add2(k<<1,l,mid,add[k]);
    Add2(k<<1|1,mid+1,r,add[k]);
    add[k]=0;
}

void modify1(int k,int l,int r,int x,int y,int v)
{
    if(x<=l&&r<=y) return Add1(k,l,r,v);
    int mid=(l+r)>>1;
    pushdown1(k,l,r,mid);
    if(x<=mid) modify1(k<<1,l,mid,x,y,v);
    if(mid<y) modify1(k<<1|1,mid+1,r,x,y,v);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

void modify2(int k,int l,int r,int x,int y,int v)
{
    if(x<=l&&r<=y) return Add2(k,l,r,v);
    int mid=(l+r)>>1;
    pushdown2(k,l,r,mid);
    if(x<=mid) modify2(k<<1,l,mid,x,y,v);
    if(mid<y) modify2(k<<1|1,mid+1,r,x,y,v);
    sum[k]=sum[k<<1]*sum[k<<1|1];
}

long long query(int k,int l,int r,int x,int y)
{
    if(x<=l&&r<=y) return sum[k];
    int mid=(l+r)>>1;
    long long res=0;
    pushdown2(k,l,r,mid);
    if(x<=mid) res+=query(k<<1,l,mid,x,y);
    if(mid<y) res+=query(k<<1|1,mid+1,r,x,y);
    return res;
}

int main()
{
    n=read();m=read();mod=read();
    for(int i=1;i<=n;i++)
      a[i]=read();

    build(1,1,n);

    for(int i=1;i<=m;i++)
    {
        opr=read();x=read();y=read();
        if(opr==1)
        {
            v=read();
            modify1(1,1,n,x,y,v);
        }
        if(opr==2)
        {
            v=read();
            modify2(1,1,n,x,y,v);
        }
        if(opr==3)
        {
            ans=query(1,1,n,x,y);
            printf("%ld\n",ans%mod);
        }
    }

    return 0;
}

原文地址：https://www.cnblogs.com/xiaoyezi-wink/p/11105832.html