增加
实现功能-单点增加+区间增加+区间查询
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中才有变化 = 变成 +=