求助 第二个过不去 请大佬帮忙看看问题

P3372 【模板】线段树 1

felixabc @ 2024-01-22 22:45:15


#include<bits/stdc++.h>
using namespace std;
int n,m,b,ans;
int c[50005];
int a[50005];
int lary[50005];
int build(int l,int r,int s){
    if(l==r){
        a[s]=c[l];
        return 0;
    }
    int mid=(l+r)/2;
    build(l,mid,s*2);
    build(mid+1,r,s*2+1);
    a[s]=a[s*2]+a[s*2+1];
}
int add(int x,int y,int l,int r,int s,int k){
    if(x<=l&&y>=r){
        a[s]+=k*(r-l+1);
        lary[s]+=k;
        return 0;
    }
    if(y>=l&&x<=l||x<=r&&y>=r||x>=l&&y<=r){
        int mid=(l+r)/2;
        add(x,y,l,mid,s*2,k);
        add(x,y,mid+1,r,s*2+1,k);
    }
    if(l!=r){
        a[s]=a[s*2]+a[s*2+1];
    }
}
int find(int x,int y,int l,int r,int s){
    if(x<=l&&y>=r){
        ans+=a[s];
        return 0;
    }
    if(y>=l&&x<=l||x<=r&&y>=r||x>=l&&y<=r){
        int mid=(l+r)/2;
        if(lary[s]!=0){
            a[s*2]+=lary[s]*(mid-l+1);
            lary[s*2]=lary[s];
            a[s*2+1]+=lary[s]*(r-mid+1);
            lary[s*2+1]=lary[s];
            lary[s]=0;
        }
        find(x,y,l,mid,s*2);
        find(x,y,mid+1,r,s*2+1);
    }
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>c[i];
    }
    build(1,n,1);
    for(int i=1;i<=m;i++){
        int d;
        cin>>d;
        if(d==1){
            int x,y,k;
            cin>>x>>y>>k;
            add(x,y,1,n,1,k);
        }
        if(d==2){
            int x,y;
            ans=0;
            cin>>x>>y;
            find(x,y,1,n,1);
            cout<<ans<<endl;
        }
    }
    return 0;
}

by lcy666666 @ 2024-01-22 22:50:03

把数组开到4\times10^5再试试 @felixabc


by lcy666666 @ 2024-01-22 22:52:29

而且应该要开long\ long吧?


by __My0217__ @ 2024-01-22 22:55:01

  1. lary[] 能不能写成 lazy[]

  2. add() 为啥不用 void

  3. add() 时是不是要 pushdown


by __My0217__ @ 2024-01-22 22:59:18

顺便,我 C++20 把你代码扔上全神秘RE。


by felixabc @ 2024-01-23 09:12:11

@My0217 打错了 忘了 pushdown我写add里面了


by felixabc @ 2024-01-23 09:23:56

@My0217 开o2就会re


by __My0217__ @ 2024-01-23 09:24:56

@felixabc 没开也re,


by __My0217__ @ 2024-01-23 09:25:29

https://www.luogu.com.cn/record/143895141


|