1
0
Fork 0
cp-templates/trees/segtree-d.cc

25 lines
603 B
C++

constexpr int MAXN=1e5+10;
int cnt,root;
int d[MAXN*2],ls[MAXN*2],rs[MAXN*2];
void update(int s,int t,int &p,int x,int c){ // add constant c to #x
if(!p)p=++cnt; // create node
if(s==t){
d[p]+=c;
return;
}
int m=s+(t-s>>1);
if(x<=m)update(s,m,ls[p],x,c);
else update(m+1,t,rs[p],x,c);
d[p]=d[ls[p]]+d[rs[p]];
}
int query(int s,int t,int p,int l,int r){ // query range [l,r]
if(!p)return 0;
if(l<=s&&t<=r)return d[p];
int m=s+(t-s>>1),sum=0;
if(l<=m)sum+=query(s,m,ls[p],l,r);
if(r>m) sum+=query(m+1,t,rs[p],l,r);
return sum;
}