POJ 3468 线段树裸题
这些天一直在看线段树，因为临近期末，所以看得断断续续，弄得有些知识点没能理解得很透切，但我也知道不能钻牛角尖，所以配合着刷题来加深理解。
然后，这是线段树裸题，而且是最简单的区间增加与查询，我参考了ACdreamer的模板，在此基础上自己用宏定义来精简了一下代码：
1 #include<cstdio> 2 typedef long long LL; 3 #define root int rt, int l, int r 4 #define lson rt*2, l, mid 5 #define rson rt*2+1, mid+1, r 6 #define makemid int mid= (l+r)>>1 7 int a,b; 8 LL c; 9 10 struct tree { 11 LL add,sum; 12 } s[400003]; 13 14 void build(root) { 15 s[rt].add= 0; 16 if(l==r){ 17 scanf("%lld",&s[rt].sum); 18 return ; 19 } 20 makemid ; 21 build(lson); 22 build(rson); 23 s[rt].sum= s[rt*2].sum+s[rt*2+1].sum; 24 } 25 26 inline void pushdown(int rt, int len) { 27 int ls= rt*2, rs= ls+1; 28 const LL &c= s[rt].add; 29 s[ls].add += c; 30 s[rs].add += c; 31 s[ls].sum += c*(len-len/2); 32 s[rs].sum += c*(len/2); 33 s[rt].add= 0; 34 } 35 36 void update(root) { 37 if(a<=l && r<=b){ 38 s[rt].add+= c; 39 s[rt].sum+= c*(r-l+1); 40 return ; 41 } 42 if(s[rt].add) pushdown(rt,r-l+1); 43 makemid ; 44 if(a<=mid) update(lson); 45 if(b>mid) update(rson); 46 s[rt].sum= s[rt*2].sum+s[rt*2+1].sum; 47 } 48 49 LL query(root) { 50 if(a<=l && r<=b) return s[rt].sum; 51 if(s[rt].add) pushdown(rt,r-l+1); 52 makemid ; 53 LL res= 0; 54 if(a<=mid) res+= query(lson); 55 if(b>mid) res+= query(rson); 56 return res; 57 } 58 59 int main(){ 60 int n,q; 61 while(~scanf("%d%d",&n,&q)){ 62 build(1,1,n); 63 while(q--){ 64 getchar(); 65 if(getchar()=='Q'){ 66 scanf("%d%d",&a,&b); 67 printf("%lld\n",query(1,1,n)); 68 } 69 else { 70 scanf("%d%d%lld",&a,&b,&c); 71 update(1,1,n); 72 } 73 } 74 } 75 return 0; 76 }
原样的代码是：
1 #include<cstdio> 2 typedef long long LL; 3 LL c; 4 int a,b; 5 6 struct tree{ 7 LL add,sum; 8 } s[400003]; 9 10 void build(int rt, int l, int r){ 11 s[rt].add= 0; 12 if(l==r){ 13 scanf("%lld",&s[rt].sum); 14 return ; 15 } 16 int mid= (l+r)>>1; 17 build(rt*2,l,mid); 18 build(rt*2+1,mid+1,r); 19 s[rt].sum= s[rt*2].sum+s[rt*2+1].sum; 20 } 21 22 inline void pushdown(int rt, int len){ 23 if(s[rt].add){ 24 int ls= rt*2, rs= ls+1; 25 s[ls].add += s[rt].add; 26 s[rs].add += s[rt].add; 27 s[ls].sum += s[rt].add*(len-len/2); 28 s[rs].sum += s[rt].add*(len/2); 29 s[rt].add= 0; 30 } 31 } 32 33 void update(int rt, int l, int r){ 34 if(a<=l && r<=b){ 35 s[rt].add+= c; 36 s[rt].sum+= c*(r-l+1); 37 return ; 38 } 39 pushdown(rt,r-l+1); 40 int mid= (l+r)>>1; 41 if(a<=mid) update(rt*2,l,mid); 42 if(b>mid) update(rt*2+1,mid+1,r); 43 s[rt].sum= s[rt*2].sum+s[rt*2+1].sum; 44 } 45 46 LL query(int rt, int l, int r){ 47 if(a<=l && r<=b) return s[rt].sum; 48 pushdown(rt,r-l+1); 49 int mid= (l+r)>>1; 50 LL res= 0; 51 if(a<=mid) res+= query(rt*2,l,mid); 52 if(b>mid) res+= query(rt*2+1,mid+1,r); 53 return res; 54 } 55 56 int main(){ 57 int n,q; 58 while(~scanf("%d%d",&n,&q)){ 59 build(1,1,n); 60 while(q--){ 61 getchar(); 62 if(getchar()=='Q'){ 63 scanf("%d%d",&a,&b); 64 printf("%lld\n",query(1,1,n)); 65 } 66 else { 67 scanf("%d%d%lld",&a,&b,&c); 68 update(1,1,n); 69 } 70 } 71 } 72 return 0; 73 }
线段树，不断进取中~~
