线段树
参考:线段树
模板题:线段树模板
写线段树需要注意的几点:
- build和update的时候最后记得加和!(d[p]=d[p<<1]+d[p<<1|1])
- lazy标记下传后记得把父节点的清除!(laz[p]=0)
// Created by CAD
#include <bits/stdc++.h>
#define lson (p<<1)
#define rson (p<<1|1)
using namespace std;
const int maxn=1e5+5;
int d[maxn<<2],a[maxn],laz[maxn<<2];
/*d[]保存区间和 a[]保存原始数据 laz[]懒惰标记为*/
//建树
void build(int s,int t,int p) {
if(s==t) {
d[p]=a[s];
return ;
}
int m=(s+t)>>1;
build(s,m,lson),build(m+1,t,rson);
d[p]=d[lson]+d[rson];
}
//延迟标记下移
void pushdown(int s,int t,int p) {
int m=(s+t)>>1;
d[lson]+=laz[p]*(m-s+1),d[rson]+=laz[p]*(t-m);
laz[lson]+=laz[p],laz[rson]+=laz[p];
laz[p]=0;
}
//单点更新
void update(int x,int s,int t,int k,int p) {
if(s==t) {
d[p]+=k;
return ;
}
int m=(s+t)>>1;
if(x<=m) update(x,s,m,k,lson);
else update(x,m+1,t,k,rson);
d[p]=d[lson]+d[rson];
}
//区间更新
void update(int l,int r,int c,int s,int t,int p) {
if(l<=s&&t<=r) {
d[p]+=(t-s+1)*c,laz[p]+=c;
return ;
}
int m=(s+t)>>1;
if(laz[p]) pushdown(s,t,p);
if(l<=m) update(l,r,c,s,m,lson);
if(r>m) update(l,r,c,m+1,t,rson);
d[p]=d[lson]+d[rson];
}
//区间求和
int getsum(int l,int r,int s,int t,int p) {
if(l<=s&&t<=r)
return d[p];
int m=(s+t)>>1;
if(laz[p]) pushdown(s,t,p);
int sum=0;
if(l<=m) sum+=getsum(l,r,s,m,lson);
if(r>m) sum+=getsum(l,r,m+1,t,rson);
return sum;
}
int main() {
build(1,n,1);
return 0;
}
CAD加油!欢迎跟我一起讨论学习算法,QQ:1401650042