RT,在 线段树2 过了的程序交到这里来只有 60pts,不知道那里错了,求大佬帮忙调
#include<bits/stdc++.h>
#define MAXN 100010
#define lson now*2
#define rson now*2+1
using namespace std;
typedef long long ll;
struct node{
int l,r;
ll sum;
ll tag_add,tag_mul;
};
node tree[MAXN << 2];
int n,p,m;
ll val[MAXN];
void push_up(int now);
void push_down(int now);
ll query(int now,int l,int r);
void build(int now,int l,int r);
void updata_add(int now,int l,int r,ll val);
void updata_mul(int now,int l,int r,ll val);
int main(){
scanf("%d%d",&n,&p);
for(int i = 1;i <= n;i++) scanf("%lld",&val[i]);
build(1,1,n);
scanf("%d",&m);
for(int i = 1;i <= m;i++){
ll op,x,y,k;
scanf("%lld",&op);
if(op == 1){
scanf("%lld%lld%lld",&x,&y,&k);
updata_mul(1,x,y,k);
}else if(op == 2){
scanf("%lld%lld%lld",&x,&y,&k);
updata_add(1,x,y,k);
}else if(op == 3){
scanf("%lld%lld",&x,&y);
printf("%lld\n",query(1,x,y));
}
}
return 0;
}
void push_up(int now){
tree[now].sum = tree[lson].sum + tree[rson].sum;
}
void push_down(int now){
if(tree[now].tag_mul != 1){
(tree[lson].tag_mul *= tree[now].tag_mul) %= p;
(tree[rson].tag_mul *= tree[now].tag_mul) %= p;
(tree[lson].tag_add *= tree[now].tag_mul) %= p;
(tree[rson].tag_add *= tree[now].tag_mul) %= p;
(tree[lson].sum *= tree[now].tag_mul) %= p;
(tree[rson].sum *= tree[now].tag_mul) %= p;
tree[now].tag_mul = 1;
}
if(tree[now].tag_add != 0){
(tree[lson].tag_add += tree[now].tag_add) %= p;
(tree[rson].tag_add += tree[now].tag_add) %= p;
(tree[lson].sum += (tree[lson].r - tree[lson].l + 1) * tree[now].tag_add) %= p;
(tree[rson].sum += (tree[rson].r - tree[rson].l + 1) * tree[now].tag_add) %= p;
tree[now].tag_add = 0;
}
}
void build(int now,int l,int r){
tree[now].l = l; tree[now].r = r;
tree[now].tag_mul = 1; tree[now].tag_add = 0;
if(tree[now].l == tree[now].r){
tree[now].sum = val[l];
return ;
}
int mid = (l + r) >> 1;
build(lson,l,mid); build(rson,mid+1,r);
push_up(now);
}
void updata_add(int now,int l,int r,ll val){
if(tree[now].l >= l && tree[now].r <= r){
(tree[now].sum += ((tree[now].r - tree[now].l + 1) * val) % p) %= p;
(tree[now].tag_add += val) %= p;
return;
}
push_down(now);
int mid = (tree[now].l + tree[now].r) >> 1;
if(r <= mid) updata_add(lson,l,r,val);
else if(l > mid) updata_add(rson,l,r,val);
else updata_add(lson,l,mid,val), updata_add(rson,mid+1,r,val);
push_up(now);
}
void updata_mul(int now,int l,int r,ll val){
if(tree[now].l >= l && tree[now].r <= r){
(tree[now].sum *= val) %= p;
(tree[now].tag_mul *= val) %= p;
(tree[now].tag_add *= val) %= p;
return;
}
push_down(now);
int mid = (tree[now].l + tree[now].r) >> 1;
if(r <= mid) updata_mul(lson,l,r,val);
else if(l > mid) updata_mul(rson,l,r,val);
else updata_mul(lson,l,mid,val), updata_mul(rson,mid+1,r,val);
push_up(now);
}
ll query(int now,int l,int r){
if(tree[now].l >= l && tree[now].r <= r){
return tree[now].sum % p;
}
push_down(now);
int mid = (tree[now].l + tree[now].r) >> 1;
if(r <= mid) return query(lson,l,r);
else if(l > mid) return query(rson,l,r);
else return (query(lson,l,mid) + query(rson,mid+1,r)) % p;
}