全WA求调

P3372 【模板】线段树 1

define @ 2024-12-14 09:47:17

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,m;
int a[100050];

struct node{
    int L;
    int R;
    int lazytage;
    int sum;
}t[400050];

void build_tree(int k,int l,int r){
    t[k].L=l;
    t[k].R=r;
    if(l==r){
        t[k].sum=a[l];
        return ;
    }
    int mid=(l+r)/2;
    build_tree(k*2,l,mid);
    build_tree(k*2+1,mid+1,r);
    t[k].sum=t[k*2].sum+t[k*2+1].sum;
}

void pushdown(int k){
    t[k*2].lazytage+=t[k].lazytage;
    t[k*2+1].lazytage+=t[k].lazytage;
    int mid=(t[k].L+t[k].R)/2;
    t[k*2].sum+=(t[k].lazytage*(mid-t[k].L+1));
    t[k*2+1].sum+=(t[k].lazytage*(t[k].R-mid));
    t[k].lazytage=0;
}

void change(int k,int l,int r,int x){
    if(t[k].L>=l&&t[k].R<=r){
        t[k].lazytage+=x;
        t[k].sum+=x*(t[k].R-t[k].L+1);
        return ;
    }
    pushdown(k);
    int mid=(t[k].L+t[k].R)/2;
    if(l<=mid) change(k*2,l,r,x);
    if(r>mid) change(k*2+1,l,r,x);
    t[k].sum=t[k*2].sum+t[k*2+1].sum;
}

int search(int k,int l,int r){
    int ans=0;
    if(t[k].L>=l&&t[k].R<=r) return t[k].sum;
    if(t[k].R<l||t[k].L>r) return 0;
    int mid=(t[k].L+t[k].R)/2;
    if(l<=mid) ans+=search(k*2,l,r);
    if(r>mid) ans+=search(k*2+1,l,r);
    return ans;
}

signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    build_tree(1,1,n);
    int typ;
    int u,v,w;
    for(int i=1;i<=m;i++){
        cin>>typ;
        if(typ==1) {
            cin>>u>>v>>w;
            change(1,u,v,w);
        }
        else{
            cin>>u>>v;
            cout<<search(1,u,v);
            cout<<"\n";
        }
    }
    return 0;
}

求求大佬帮助本蒟蒻吧


by bijhla @ 2024-12-14 16:28:03

@define

query函数没有pushdown

具体的

int search(int k,int l,int r){
    int ans=0;
    if(t[k].L>=l&&t[k].R<=r) return t[k].sum;
    if(t[k].R<l||t[k].L>r) return 0;
    int mid=(t[k].L+t[k].R)/2;
    pushdown(k);//加上这个
    if(l<=mid) ans+=search(k*2,l,r);
    if(r>mid) ans+=search(k*2+1,l,r);
    return ans;
}

by define @ 2024-12-14 19:04:27

@bijhla

感谢大佬


|