line tree(不知道对不对)求助

P3372 【模板】线段树 1

CSP_JAKME @ 2024-10-18 21:43:07

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int n,m,a[maxn],tree[maxn*4];
int lzy[maxn*4]; 
void pushup(int u){
    tree[u]=tree[u*2]+tree[u*2+1];
}
void build(int u,int left,int right){
    if(left==right){
        tree[u]=a[left];
        return;
    }
    int mid=(left+right)/2;
    build(u*2,left,mid);
    build(u*2+1,mid+1,right);
    pushup(u);
    return;
}
bool intheline(int L,int R,int l,int r){
    return l<=L&&r>=R;
}
bool cdline(int L,int R,int l,int r){
    return L<=r&&R>=l;
}
int sum(int u,int left,int right,int l,int r){
    if(intheline(left,right,l,r)) return tree[u];
    else if(cdline(left,right,l,r)){
        int mid=(left+right)/2;
        return sum(u*2,left,mid,l,r)+sum(u*2+1,mid+1,right,l,r);
    }
    return 0;
}
void make(int u,int k,int len){
    lzy[u]+=k;
    tree[u]+=len*k;
    return;
}
void pushdown(int u,int left,int right){
    int mid=(left+right)/2;
    make(u*2,lzy[u],mid-left+1);
    make(u*2+1,lzy[u],right-mid);
    lzy[u]=0;
    return;
}
/*
void printt(int u){
    if(u>2*n-1) return;
    cout << tree[u] << " ";
    printt(u*2);
    printt(u*2+1);
}
*/
void update(int u,int left,int right,int l,int r,int k){
    if(intheline(left,right,l,r)){
        make(u,k,right-left+1);
        return;
    }else if(cdline(left,right,l,r)){
        int mid=(left+right)/2;
        pushdown(u,left,right);
        update(u*2,left,mid,l,r,k);
        update(u*2+1,mid+1,right,l,r,k);
        pushup(u);
    }else return;
}
int main(){
    cin >> n >> m;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    build(1,1,n);
    for(int i=1;i<=m;i++){
        int opt,x,y;
        cin >> opt >> x >> y;
        if(opt==1){
            int k;
            cin >> k;
            update(1,1,n,x,y,k);
        }else{
            printf("%d\n",sum(1,1,n,x,y));
        }
    }
    return 0;
}

样例第4个操作直接改了根节点


by CSP_JAKME @ 2024-10-18 22:08:28

@yanyan2 不行的


by yanyan2 @ 2024-10-18 22:12:56

@CSP_JAKME 什么意思


by CSP_JAKME @ 2024-10-18 22:14:14

@yanyan2 你改了[1,5].你查询[1,4],可是你只改了[1,5]


by yanyan2 @ 2024-10-18 22:20:05

@CSP_JAKME 不是你会不会 lazytag 啊


by CSP_JAKME @ 2024-10-18 22:28:58

@yanyan2 所以哪错了


by yanyan2 @ 2024-10-19 07:00:04

@CSP_JAKME 就是直接改整个点,你查询的时候分开来查每个区间的信息不就对了吗


上一页 |