增加
实现功能-单点增加+区间增加+区间查询
struct SegTree
{
ll Sum[maxn<<2],Max[maxn<<2],cnt[maxn<<2],lz[maxn<<2];
void init()
{
memset(Sum,0,sizeof(Sum));
memset(Max,0,sizeof(Max));
memset(cnt,0,sizeof(cnt));
memset(lz,0,sizeof(lz));
}
void push_up(int rt) /// 上更新函数
{
Sum[rt] = Sum[rt<<1] + Sum[rt<<1|1];
Max[rt] = max(Max[rt<<1],Max[rt<<1|1]);
cnt[rt] = cnt[rt<<1] + cnt[rt<<1|1];
}
void push_down(int rt) /// 单点(区间)增加的下更新函数
{
if(!lz[rt]) return ;
lz[rt<<1] += lz[rt];
lz[rt<<1|1] += lz[rt];
Sum[rt<<1] += lz[rt]*cnt[rt<<1];
Sum[rt<<1|1] += lz[rt]*cnt[rt<<1|1];
Max[rt<<1] += lz[rt];
Max[rt<<1|1] += lz[rt];
lz[rt] = 0;
}
void build(ll arr[],int l,int r,int rt) /// 数组建立线段树
{
if(l == r) {
Max[rt] = arr[l];
Sum[rt] = arr[l];
cnt[rt] = 1;
lz [rt] = 0;
return;
}
int mid = (l + r) >> 1;
build(arr,l,mid,rt<<1);
build(arr,mid+1,r,rt<<1|1);
push_up(rt);
}
void activate(int pos,ll val,int l,int r,int rt) /// 单点建立线段树
{
if(l == r) {
Sum[rt] += val;
Max[rt] += val;
cnt[rt] ++;
return ;
}
push_down(rt);
int mid = (l + r) >> 1;
if(pos <= mid) activate(pos,val,l,mid,rt<<1);
else activate(pos,val,mid+1,r,rt<<1|1);
push_up(rt);
}
void update(int pos,ll val,int l,int r,int rt) /// 单点修改(增加)
{
if(l == r) {
Sum[rt] += val; /// 增加
Max[rt] += val; ///
lz[rt] += val; ///
return;
}
push_down(rt);
int mid = (l + r) >> 1;
if(pos <= mid) update(pos,val,l,mid,rt<<1);
else update(pos,val,mid+1,r,rt<<1|1);
push_up(rt);
}
void update(int ql,int qr,ll val,int l,int r,int rt) /// 区间修改(增加)
{
if(ql == l && qr == r) {
Sum[rt] += val*cnt[rt]; /// 增加
Max[rt] += val; ///
lz[rt] += val; ///
return;
}
push_down(rt);
int mid = (l + r) >> 1;
if(qr <= mid) update(ql,qr,val,l,mid,rt<<1);
else if(ql > mid) update(ql,qr,val,mid+1,r,rt<<1|1);
else {
update(ql,mid,val,l,mid,rt<<1);
update(mid+1,qr,val,mid+1,r,rt<<1|1);
}
push_up(rt);
}
ll Max_query(int ql,int qr,int l,int r,int rt) /// 区间最大值查询
{
if(ql == l && qr == r) {
return Max[rt];
}
push_down(rt);
int mid = (l + r) >> 1;
if(qr <= mid) return Max_query(ql,qr,l,mid,rt<<1);
if(ql > mid) return Max_query(ql,qr,mid+1,r,rt<<1|1);
return max(Max_query(ql,mid,l,mid,rt<<1),Max_query(mid+1,qr,mid+1,r,rt<<1|1));
}
ll Sum_query(int ql,int qr,int l,int r,int rt) /// 区间求和查询
{
if(ql == l && qr == r) {
return Sum[rt];
}
push_down(rt);
int mid = (l + r) >> 1;
if(qr <= mid) return Sum_query(ql,qr,l,mid,rt<<1);
if(ql > mid) return Sum_query(ql,qr,mid+1,r,rt<<1|1);
return Sum_query(ql,mid,l,mid,rt<<1) + Sum_query(mid+1,qr,mid+1,r,rt<<1|1);
}
}seg;
实现功能-单点覆盖+区间覆盖+区间查询
struct SegTree
{
ll Sum[maxn<<2],Max[maxn<<2],cnt[maxn<<2],lz[maxn<<2];
void init()
{
memset(Sum,0,sizeof(Sum));
memset(Max,0,sizeof(Max));
memset(cnt,0,sizeof(cnt));
memset(lz,0,sizeof(lz));
}
void push_up(int rt) /// 上更新函数
{
Sum[rt] = Sum[rt<<1] + Sum[rt<<1|1];
Max[rt] = max(Max[rt<<1],Max[rt<<1|1]);
cnt[rt] = cnt[rt<<1] + cnt[rt<<1|1];
}
void push_down(int rt) /// 单点(区间)覆盖的下更新函数
{
if(!lz[rt]) return ;
lz[rt<<1] = lz[rt];
lz[rt<<1|1] = lz[rt];
Sum[rt<<1] = lz[rt]*cnt[rt<<1];
Sum[rt<<1|1] = lz[rt]*cnt[rt<<1|1];
Max[rt<<1] = lz[rt];
Max[rt<<1|1] = lz[rt];
lz[rt] = 0;
}
void build(ll arr[],int l,int r,int rt) /// 数组建立线段树
{
if(l == r) {
Max[rt] = arr[l];
Sum[rt] = arr[l];
cnt[rt]++;
return;
}
push_down(rt);
int mid = (l + r) >> 1;
build(arr,l,mid,rt<<1);
build(arr,mid+1,r,rt<<1|1);
push_up(rt);
}
void activate(int pos,ll val,int l,int r,int rt) /// 单点建立线段树
{
if(l == r) {
Sum[rt] += val;
Max[rt] += val;
cnt[rt] ++;
return ;
}
push_down(rt);
int mid = (l + r) >> 1;
if(pos <= mid) activate(pos,val,l,mid,rt<<1);
else activate(pos,val,mid+1,r,rt<<1|1);
push_up(rt);
}
void update(int pos,ll val,int l,int r,int rt) /// 单点修改(覆盖)
{
if(l == r) {
Sum[rt] = val; /// 和增加不同的地方
Max[rt] = val; ///
lz[rt] = val; ///
return;
}
push_down(rt);
int mid = (l + r) >> 1;
if(pos <= mid) update(pos,val,l,mid,rt<<1);
else update(pos,val,mid+1,r,rt<<1|1);
push_up(rt);
}
void update(int ql,int qr,ll val,int l,int r,int rt) /// 区间修改(覆盖)
{
if(ql == l && qr == r) {
Sum[rt] = val*cnt[rt]; /// 和增加不同的地方
Max[rt] = val; ///
lz[rt] = val; ///
return;
}
push_down(rt);
int mid = (l + r) >> 1;
if(qr <= mid) update(ql,qr,val,l,mid,rt<<1);
else if(ql > mid) update(ql,qr,val,mid+1,r,rt<<1|1);
else {
update(ql,mid,val,l,mid,rt<<1);
update(mid+1,qr,val,mid+1,r,rt<<1|1);
}
push_up(rt);
}
ll Max_query(int ql,int qr,int l,int r,int rt) /// 区间最大值查询
{
if(ql == l && qr == r) {
return Max[rt];
}
push_down(rt);
int mid = (l + r) >> 1;
if(qr <= mid) return Max_query(ql,qr,l,mid,rt<<1);
if(ql > mid) return Max_query(ql,qr,mid+1,r,rt<<1|1);
return max(Max_query(ql,mid,l,mid,rt<<1),Max_query(mid+1,qr,mid+1,r,rt<<1|1));
}
ll Sum_query(int ql,int qr,int l,int r,int rt) /// 区间求和查询
{
if(ql == l && qr == r) {
return Sum[rt];
}
push_down(rt);
int mid = (l + r) >> 1;
if(qr <= mid) return Sum_query(ql,qr,l,mid,rt<<1);
if(ql > mid) return Sum_query(ql,qr,mid+1,r,rt<<1|1);
return Sum_query(ql,mid,l,mid,rt<<1) + Sum_query(mid+1,qr,mid+1,r,rt<<1|1);
}
}seg;
注意到:
覆盖和增加
只有在push_down()函数和update中才有变化 = 变成 +=