线段树

// 结构体版
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

#define N 100005
#define ll long long
#define lc u<<1
#define rc u<<1|1
ll w[N];
struct Tree{ //线段树
  ll l,r,sum,add;
}tr[N*4];

void pushup(ll u){ //上传
  tr[u].sum=tr[lc].sum+tr[rc].sum;
}
void pushdown(ll u){ //下传
  if(tr[u].add){
    tr[lc].sum+=tr[u].add*(tr[lc].r-tr[lc].l+1),
    tr[rc].sum+=tr[u].add*(tr[rc].r-tr[rc].l+1),
    tr[lc].add+=tr[u].add,
    tr[rc].add+=tr[u].add,
    tr[u].add=0;      
  }
}
void build(ll u,ll l,ll r){ //建树
  tr[u]={l,r,w[l],0};
  if(l==r) return;
  ll m=l+r>>1;
  build(lc,l,m);
  build(rc,m+1,r);
  pushup(u);
}
void change(ll u,ll l,ll r,ll k){ //区修
  if(l<=tr[u].l&&tr[u].r<=r){
    tr[u].sum+=(tr[u].r-tr[u].l+1)*k;
    tr[u].add+=k;
    return;
  }
  ll m=tr[u].l+tr[u].r>>1;
  pushdown(u);
  if(l<=m) change(lc,l,r,k);
  if(r>m) change(rc,l,r,k);
  pushup(u);
}
ll query(ll u,ll l,ll r){ //区查
  if(l<=tr[u].l && tr[u].r<=r) return tr[u].sum;
  ll m=tr[u].l+tr[u].r>>1;
  pushdown(u);
  ll sum=0;
  if(l<=m) sum+=query(lc,l,r);
  if(r>m) sum+=query(rc,l,r);
  return sum;
}
int main(){
  ll n,m,op,x,y,k;  
  cin>>n>>m;
  for(int i=1; i<=n; i ++) cin>>w[i];
  
  build(1,1,n);
  while(m--){
    cin>>op>>x>>y;
    if(op==2)cout<<query(1,x,y)<<endl;
    else cin>>k,change(1,x,y,k);
  }
  return 0;
}

posted on 2025-05-04 11:05  下头小美  阅读(10)  评论(0)    收藏  举报