贴一个自己实现线段树单点修改,区间查询,区间修改的模板
struct Node{
int l, r;
int sum, minn, maxx; //特征值
int lazy;
}tree[maxn<<2]; //开四倍空间
int num[maxn]; //数值数组
void build(int i, int l, int r){
tree[i].l = l;
tree[i].r = r;
if(l == r){
tree[i].sum = num[l];
tree[i].minn = num[l];
tree[i].maxx = num[l];
tree[i].lazy = 0;
return ;
}
int mid = (l+r)>>1;
build(i<<1, l, mid);
build(i<<1|1, mid+1, r); //子树建树后更新特征值
tree[i].sum = tree[i<<1].sum + tree[i<<1|1].sum;
tree[i].maxx = max(tree[i<<1].maxx, tree[i<<1|1].maxx);
tree[i].minn = min(tree[i<<1].minn, tree[i<<1|1].minn);
tree[i].lazy = 0;
return ;
}
void push_down(int i){ //以lazy表示区间加为例更改特征值
if(tree[i].lazy){
tree[i<<1].sum += tree[i].lazy * (tree[i<<1].r-tree[i<<1].l+1);
tree[i<<1].maxx += tree[i].lazy;
tree[i<<1].minn += tree[i].lazy;
tree[i<<1|1].sum += tree[i].lazy * (tree[i<<1|1].r-tree[i<<1|1].l+1);
tree[i<<1|1].maxx += tree[i].lazy;
tree[i<<1|1].minn += tree[i].lazy;
tree[i<<1].lazy += tree[i].lazy;
tree[i<<1|1].lazy += tree[i].lazy;
tree[i].lazy = 0;
}
}
void addx(int i, int x, int k){ // num[x] += k
tree[i].sum += k;
if(tree[i].l == tree[i].r) return ;
int mid = (tree[i].l+tree[i].r)>>1;
if(x <= mid) addx(i<<1, x, k);
else addx(i<<1|1, x, k);
/** sum可以在下放前更新, maxx, minn 需要递归更新*/
tree[i].maxx = max(tree[i<<1].maxx, tree[i<<1|1].maxx);
tree[i].minn = min(tree[i<<1].minn, tree[i<<1|1].minn);
}
void addlr(int i, int l, int r, int k){ // num[l~r] += k
if(tree[i].l==l && tree[i].r==r){
tree[i].sum += k*(tree[i].r-tree[i].l+1);
tree[i].maxx += k;
tree[i].minn -= k;
tree[i].lazy += k;
return ;
}
push_down(i);
int mid = (tree[i].l+tree[i].r)>>1;
if(mid >= r)
addlr(i<<1, l, r, k);
else if(mid < l)
addlr(i<<1|1, l, r, k);
else {
addlr(i<<1, l, mid, k);
addlr(i<<1|1, mid+1, r, k);
}
tree[i].sum = tree[i<<1].sum + tree[i<<1|1].sum;
tree[i].maxx = max(tree[i<<1].maxx, tree[i<<1|1].maxx);
tree[i].minn = min(tree[i<<1].minn, tree[i<<1|1].minn);
}
int query(int i, int l, int r){ //以sum为例
if(tree[i].l==l && tree[i].r==r)
return tree[i].sum;
push_down(i);
int mid = (tree[i].l+tree[i].r)>>1;
if(mid >= r)
return query(i<<1, l, r);
else if(mid < l)
return query(i<<1|1, l, r);
else
return query(i<<1, l, mid) + query(i<<1|1, mid+1, r);
}
u p d a t e 版 update版 update版
struct Node{
int l, r;
int sum, minn, maxx; //特征值
int lazy;
}tree[maxn<<2]; //开四倍空间
int num[maxn]; //数值数组
void update(int i){
tree[i].sum = tree[i<<1].sum + tree[i<<1|1].sum;
tree[i].maxx = max(tree[i<<1].maxx, tree[i<<1|1].maxx);
tree[i].minn = min(tree[i<<1].minn, tree[i<<1|1].minn);
}
void build(int i, int l, int r){
tree[i].l = l;
tree[i].r = r;
if(l == r){
tree[i].sum = num[l];
tree[i].minn = num[l];
tree[i].maxx = num[l];
tree[i].lazy = 0;
return ;
}
int mid = (l+r)>>1;
build(i<<1, l, mid);
build(i<<1|1, mid+1, r); //子树建树后更新特征值
update(i); //向上更新 该节点所有值从子节点得到 无法更新lazy和映射lazy状态
tree[i].lazy = 0;
return ;
}
void push_down(int i){ //以lazy表示区间加为例更改特征值
if(tree[i].lazy){
tree[i<<1].sum += tree[i].lazy * (tree[i<<1].r-tree[i<<1].l+1);
tree[i<<1].maxx += tree[i].lazy;
tree[i<<1].minn += tree[i].lazy;
tree[i<<1|1].sum += tree[i].lazy * (tree[i<<1|1].r-tree[i<<1|1].l+1);
tree[i<<1|1].maxx += tree[i].lazy;
tree[i<<1|1].minn += tree[i].lazy;
tree[i<<1].lazy += tree[i].lazy;
tree[i<<1|1].lazy += tree[i].lazy;
tree[i].lazy = 0;
}
}
void addx(int i, int x, int k){ // num[x] += k
tree[i].sum += k;
if(tree[i].l == tree[i].r) return ;
int mid = (tree[i].l+tree[i].r)>>1;
if(x <= mid) addx(i<<1, x, k);
else addx(i<<1|1, x, k);
/** sum可以在下放前更新, maxx, minn 需要递归更新*/
tree[i].maxx = max(tree[i<<1].maxx, tree[i<<1|1].maxx);
tree[i].minn = min(tree[i<<1].minn, tree[i<<1|1].minn);
}
void addlr(int i, int l, int r, int k){ // num[l~r] += k
if(tree[i].l==l && tree[i].r==r){
tree[i].sum += k*(tree[i].r-tree[i].l+1);
tree[i].maxx += k;
tree[i].minn -= k;
tree[i].lazy += k;
return ;
}
push_down(i);
int mid = (tree[i].l+tree[i].r)>>1;
if(mid >= r)
addlr(i<<1, l, r, k);
else if(mid < l)
addlr(i<<1|1, l, r, k);
else {
addlr(i<<1, l, mid, k);
addlr(i<<1|1, mid+1, r, k);
}
update(i);
}
int query(int i, int l, int r){ //以sum为例
if(tree[i].l==l && tree[i].r==r)
return tree[i].sum;
push_down(i);
int mid = (tree[i].l+tree[i].r)>>1;
if(mid >= r)
return query(i<<1, l, r);
else if(mid < l)
return query(i<<1|1, l, r);
else
return query(i<<1, l, mid) + query(i<<1|1, mid+1, r);
}