线段树 区间求和模板 (区间修改)
2017-08-30 23:58
323 查看
#include <stdio.h> #include <iostream> #include <algorithm> #include <math.h> #include <cstring> #include <queue> #include <map> #include <vector> #include <string> #define mem(a) memset(a,0,sizeof(a)) #define mem2(a) memset(a,-1,sizeof(a)) #define mod 1000000007 #define mx 100005 using namespace std; long long a[mx],sum[mx<<2],add[mx<<2];//直接开long long 就好了。sum和add数组要开数据量的四倍别忘了。 int n,l; void pushup (int x)//维护当前线段的区间和。 { sum[x]=sum[x<<1]+sum[x<<1|1]; } void build (int l,int r,int x)//建树 { if(l==r) { sum[x]=a[l]; return; } int mid = (l+r)/2; build(l,mid,x<<1); build(mid+1,r,x<<1|1); pushup(x); } void pushdown (int x,int lx,int rx)//下推懒惰标记。 { //lx,rx分别是左子树,右子树的数量。 if(add[x]) { //下推标记 add[x<<1]+=add[x]; add[x<<1|1]+=add[x]; //修改子节点的sum使之与对应的add相对应。 sum[x<<1]+=add[x]*lx; sum[x<<1|1]+=add[x]*rx; //清除该节点的懒惰标记。 add[x]=0; 4000 } } void update (int L,int R,int C,int l,int r,int x) { if(L<=l&&r<=R) { sum[x]+=C*(r-l+1); add[x]+=C; return ; } int mid=(l+r)/2; pushdown(x,mid-l+1,r-mid); //一定要在向下更新操作进行之前下推懒惰标记。 if(L<=mid) update(L,R,C,l,mid,x<<1); if(R>mid) update(L,R,C,mid+1,r,x<<1|1); pushup(x); } long long query (int L,int R,int l,int r,int x)//返回值一定要写long long 别问我怎么知道的。 { if(L<=l&&r<=R) { return sum[x]; } int mid = (l+r)>>1; pushdown(x,mid-l+1,r-mid); //下推懒惰标记,更新左子树和右子树维护的线段的区间和的值。 long long ans=0; if(L<=mid) ans+=query(L,R,l,mid,x<<1); if(R>mid ) ans+=query(L,R,mid+1,r,x<<1|1); return ans; } int main() { #ifndef ONLINE_JUDGE freopen("1.txt","r",stdin); #endif // ONLINE_JUDGE ios_base::sync_with_stdio(false); cin.tie(0); char c; int x,y,z; while(cin>>n>>l) { mem(sum); mem(add); mem(a); for(int i=1; i<=n; ++i) cin>>a[i]; build(1,n,1); while(l--) { cin>>c; if(c=='Q') { cin>>x>>y; cout<<query(x,y,1,n,1)<<endl; } if(c=='C') { cin>>x>>y>>z; update(x,y,z,1,n,1); } } } return 0; }
相关文章推荐
- hdu 1166 线段树 单点修改 + 询问区间求和 (线段树模板)
- 敌兵布阵 HDU - 1166 (线段树单点修改区间求和模板)
- 【模板】线段树区间修改、区间求和、查询最值
- 【模板】线段树_区间最值、区间求和、修改
- 算法模板——线段树1(区间加法+区间求和)
- [kuangbin带你飞]专题七 线段树 A - 敌兵布阵 (单点修改,区间求和)
- 线段树区间修改与查询(求和)
- 线段树模板(区间修改)——hdu1698
- 线段树经典操作模板(单点更新,替换;区间更新,替换;区间求和求最值)
- codevs 1082 线段树练习 3(区间修改+区间求和)
- 非递归线段树区间修改区间求和的两种实现(以POJ 3468为例)
- hdu 1698 线段树的区间修改 模板~~
- HDU - 3577 Fast Arrangement (线段树区间修改及查询模板题)
- 线段树 --- (单点更新、区间求和、模板题)
- 操作格子 (线段树)(单点修改,区间求最值,区间求和)
- 基本线段树模板(建树、点/区间修改、查询)
- hihocoder 1078 线段树的区间修改线段树(区间修改 区间求和)
- [ 模板 ] 线段树区间修改
- hdu1394 Minimum Inversion Number(线段树单点修改+区间求和)
- poj 3468 A Simple Problem with Integers(线段树区间修改+区间求和)