求调

P3372 【模板】线段树 1

Austin_Chen @ 2024-12-27 19:05:23

RT,样例都没过

#include<bits/stdc++.h>
#define int long long 
using namespace std;
const int N=100000; 
int a[N*4+10],w[N*4+10];
int lzy[N*4+10]; 
int n,m,x,y,k,op;
void pushup(const int u){w[u]=w[u*2]+w[u*2+1];}
void build(const int u,int L,int R)
{
    if(L==R)
    {
        w[u]=a[L];
        return;
    }
    int Mid=(L+R)/2;
    build(u*2,L,Mid);
    build(u*2+1,Mid+1,R);
    pushup(u);
}
int query_o(int u,int L,int R,int p)
{
    if(L==R) return w[u];
      else 
      {
        int Mid=(L+R)/2;
        if(Mid>=p) query_o(u*2,L,Mid,p);
          else query_o(u*2,Mid+1,R,p);
      }
}
void update_o(int u,int L,int R,int p,int o)
{
    if(L==R) w[u]=o;
      else 
      {
        int Mid=(L+R)/2;
        if(Mid>=p) update_o(u*2,L,Mid,p,o);
          else update_o(u*2+1,Mid+1,R,p,o);
        pushup(u); 
      }
}
bool InRange(int L,int R,int l,int r){ return (l<=L)&&(R<=r);}
bool OutofRange(int L,int R,int l,int r){ return (L>r)||(R>l);}
void maketag(int u,int len,int o)
{
    lzy[u]+=o;
    w[u]+=len*o; 
}
void pushdown(int u,int L,int R)
{
    int Mid=(L+R)/2;
    maketag(u*2,Mid-L+1,lzy[u]);
    maketag(u*2+1,R-Mid,lzy[u]);
    lzy[u]=0; 
}
int query(int u,int L,int R,int l,int r)
{
    if(InRange(L,R,l,r)) return w[u];
      else if(!OutofRange(L,R,l,r))
      {
        int Mid=(L+R)/2;
        return query(u*2,L,Mid,l,r)+query(u*2+1,Mid+1,R,l,r);
      }
        else return 0;
}
void update(int u,int L,int R,int l,int r,int o)
{
    if(InRange(L,R,l,r)) maketag(u,R-L+1,o);
      else if(!OutofRange(L,R,l,r))
      {
        int Mid=(L+R)/2;
        pushdown(u,L,R);
        update(u*2,L,Mid,l,r,o);
        update(u*2+1,Mid+1,R,l,r,o);
        pushup(u);
      }
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        cin>>op;
        if(op==1) 
        {
            cin>>x>>y>>k;
            update(1,1,n,x,y,k);
        }
          else
          {
            cin>>x>>y;
            cout<<query(1,1,n,x,y)<<'\n';
          }
    }
    return 0;
}

by CSP_JAKME @ 2024-12-27 19:52:16

在query的时候也要pushdown


by Austin_Chen @ 2025-01-11 08:35:25

@CSP_JAKME谢谢,已关


by Austin_Chen @ 2025-01-11 08:38:19

@CSP_JAKME但似乎还是不对


|