线段树求调
查看原帖
线段树求调
648807
L_life楼主2022/10/11 08:34
#include<bits/stdc++.h>
#define int long long
using namespace std;
int mod;
struct ma{
    int le,re;
    int sum,lazy,cheng;
}sa[400000];
long long a[400000],n,m;
void build(int p,int l,int r){
    sa[p].le=l,sa[p].re=r;
    sa[p].cheng=1;
    if(l==r)
    {
        sa[p].sum=a[l]%mod;
        return ;
    }
    int mid=(l+r)/2;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    sa[p].sum=sa[p*2].sum+sa[p*2+1].sum;
} 
void pushup(int p)
{
    sa[p*2].sum=(sa[p*2].sum*sa[p].cheng+sa[p].lazy*(sa[p*2].re-sa[p*2+1].le+1))%mod;
    sa[p*2+1].sum=(sa[p*2+1].sum*sa[p].cheng+sa[p].lazy*(sa[p*2+1].re-sa[p*2+1].le+1))%mod;
    sa[p*2].cheng=(sa[p*2].cheng*sa[p].cheng)%mod;
    sa[p*2+1].cheng=(sa[p*2+1].cheng*sa[p].cheng)%mod;
    sa[p*2].lazy=(sa[p*2].lazy*sa[p].cheng+sa[p].lazy)%mod;
    sa[p*2+1].lazy=(sa[p*2+1].lazy*sa[p].cheng+sa[p].lazy)%mod;
    sa[p].lazy=0,sa[p].cheng=1;
}
void change(int p,int l,int r,int z)
{
	if(r<sa[p].le||l>sa[p].re) return ;
    if(r>=sa[p].re&&l<=sa[p].le)
    {
        sa[p].sum+=z*(sa[p].re-sa[p].le+1)%mod;
        sa[p].lazy+=z;
        return ;
    }
    pushup(p);
    sa[p].sum=sa[p*2].sum+sa[p*2+1].sum;
    int mid=(sa[p].re+sa[p].le)/2;
    if(l<=mid) change(p*2,l,r,z);
    if(r>mid) change(p*2+1,l,r,z);
    sa[p].sum=sa[p*2].sum+sa[p*2+1].sum;
}
void change1(int p,int l,int r,int z)
{
	if(r<sa[p].le||l>sa[p].re) return ;
    if(r>=sa[p].re&&l<=sa[p].le)
    {
        sa[p].lazy=(sa[p].lazy*z)%mod;
        sa[p].cheng=(sa[p].cheng*z)%mod;
        sa[p].sum=(sa[p].sum*z)%mod;
        return ;
    }
    pushup(p);
    sa[p].sum=sa[p*2].sum+sa[p*2+1].sum;
    int mid=(sa[p].re+sa[p].le)/2;
    if(l<=mid) change1(p*2,l,r,z);
    if(r>mid) change1(p*2+1,l,r,z);
    sa[p].sum=sa[p*2].sum+sa[p*2+1].sum;
}
int answer(int p,int l,int r)
{
    if(r>=sa[p].re&&l<=sa[p].le)
    {
        return sa[p].sum;
    }
    pushup(p);
    int ans=0;
    int mid=(sa[p].le+sa[p].re)/2;
    if(l<=mid) ans=(ans+answer(p*2,l,r))%mod;
    if(r>mid) ans=(ans+answer(p*2+1,l,r))%mod;
    return ans;
}
signed main()
{
    cin>>n>>m>>mod;
    int t;
    for(int i=1;i<=n;++i) cin>>a[i];
    build(1,1,n);
    for(int i=1;i<=m;++i)
    {
        cin>>t;
        if(t==1)
        {
        	int x,y,z;
            cin>>x>>y>>z;
            change1(1,x,y,z);
        }
        else if(t==2){
        	int x,y,z;
            cin>>x>>y>>z;
            change(1,x,y,z);
        }
        else {
        	int x,y;
            cin>>x>>y;
            cout<<answer(1,x,y)%mod<<endl;
        }
    }
}
2022/10/11 08:34
加载中...