线段树求调
查看原帖
线段树求调
547908
NightTide楼主2022/5/10 22:06

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;
}
2022/5/10 22:06
加载中...