洛谷P3373 线段树2

时间:2022-12-17 20:00:50

毒瘤题。找了一下午+晚上的BUG,才发现原来query_tree写的是a%p;

真的是一个教训

#include<iostream>
#include<cmath> 
#include<cstdio>
#include<cstring>
#include<queue>
#define lson i*2,l,mid
#define rson i*2+1,mid+1,r
using namespace std;

struct tree{
long long mul;
long long add;
long long sum;
int l,r; 
}t[400860];

int n,m,a[100860],p;

void build_tree(int i,int l,int r)
{
t[i].l=l;
t[i].r=r;
t[i].mul=1;
t[i].sum=0;
t[i].add=0; 
if(l==r)
{
t[i].sum=a[l];
return ;
}
int mid=(l+r)/2;
build_tree(lson);
build_tree(rson);
t[i].sum=(t[i*2].sum+t[i*2+1].sum)%p;
}

void pushdown2(int i)
{
t[i*2].mul=t[i].mul*t[i*2].mul%p;
t[i*2+1].mul=t[i].mul*t[i*2+1].mul%p;
t[i*2].add=(t[i*2].add*t[i].mul+t[i].add)%p;
t[i*2+1].add=(t[i*2+1].add*t[i].mul+t[i].add)%p;
t[i*2].sum=(t[i].mul*t[i*2].sum+t[i].add*(t[i*2].r-t[i*2].l+1))%p;
t[i*2+1].sum=(t[i].mul*t[i*2+1].sum+t[i].add*(t[i*2+1].r-t[i*2+1].l+1))%p;
t[i].add=0;
t[i].mul=1;
}
void pushdown(int i)
{
t[i*2].mul=t[i].mul*t[i*2].mul%p;
t[i*2+1].mul=t[i*2+1].mul*t[i].mul%p;
t[i*2].add=(t[i*2].add*t[i].mul+t[i].add)%p;
t[i*2+1].add=(t[i*2+1].add*t[i].mul+t[i].add)%p;
t[i*2].sum=(t[i].mul*t[i*2].sum+t[i].add*(t[i*2].r-t[i*2].l+1))%p;
t[i*2+1].sum=(t[i].mul*t[i*2+1].sum+t[i].add*(t[i*2+1].r-t[i*2+1].l+1))%p;
t[i].add=0;
t[i].mul=1;
}
void mul_tree(int i,int l,int r,int x,int y,int a)
{
if(l>=x&&r<=y)
{
t[i].sum*=a%p;
t[i].mul*=a%p;
t[i].add*=a%p;
return ;
}
pushdown(i);
int mid=(l+r)/2;
if(x<=mid) mul_tree(lson,x,y,a);
if(y>mid) mul_tree(rson,x,y,a);
t[i].sum=(t[i*2].sum+t[i*2+1].sum)%p;
}
void add_tree(int i,int l,int r,int x,int y,int a)
{
if(l>=x&&r<=y)
{
//t[i].mul*=a%p;    
t[i].add+=a%p;
t[i].sum+=(t[i].r-t[i].l+1)*a%p;
return ;
}
pushdown(i);
int mid=(l+r)/2;
if(x<=mid) add_tree(lson,x,y,a);
if(y>mid) add_tree(rson,x,y,a);
t[i].sum=(t[i*2].sum+t[i*2+1].sum)%p;
} 

int query_tree(int i,int l,int r,int a,int b)
{
if(l>=a&&r<=b)
{
return t[i].sum%p;
}
int mid=(l+r)/2;
long long ans=0;
pushdown(i);
if(a<=mid) ans+=query_tree(lson,a,b)%p;
if(b>mid) ans+=query_tree(rson,a,b)%p;
return ans%p;

}
int main()
{
scanf("%d %d %d",&n,&m,&p);
int i,j;
for(i=1;i<=n;i++)
scanf("%d",&a[i]); 
build_tree(1,1,n);
for(i=1;i<=m;i++)
{
int k;
int t1,t2,t3;
scanf("%d",&k);
if(k==1)
{
scanf("%d %d %d",&t1,&t2,&t3);
mul_tree(1,1,n,t1,t2,t3);    
//printf("%d \n",query_tree(1,1,n,t1,t2)%p);
}
if(k==2)
{
scanf("%d %d %d",&t1,&t2,&t3);
add_tree(1,1,n,t1,t2,t3);
//printf("%d \n",query_tree(1,1,n,t1,t2)%p);
}
if(k==3)
{
scanf("%d %d",&t1,&t2);
printf("%d \n",query_tree(1,1,n,t1,t2)%p);
}
} 
return 0;
}