poj 3468

  1. #include <vector>
  2. #include <queue>
  3. #include <stack>
  4. #include <set>
  5. #include <map>
  6. #include <algorithm>
  7. #include <iostream>
  8. #include <cstdio>
  9. #include <cstring>
  11. #define MAXN 100005
  12. #define LC(x) ((x) << 1) 
  13. #define RC(x) ((x) << 1 | 1) 
  14. #define Mid(x,y) (((x) + (y)) >> 1)
  15. #define Min(a,b) ((a) < (b) ? (a) : (b))
  16. #define Max(a,b) ((a) > (b) ? (a) : (b))
  18. long long int sum[MAXN<<2],dlt[MAXN<<2];
  19. int lft[MAXN<<2],rht[MAXN<<2];
  20. long long int init_v[MAXN+10];
  22. using namespace std;
  24.  
  26. void pushdown(int rt){
  27.     if(dlt[rt]){
  28.         dlt[LC(rt)]+=dlt[rt];
  29.         sum[LC(rt)]+=(rht[LC(rt)]-lft[LC(rt)]+1)*dlt[rt]; //width*height
  30.         dlt[RC(rt)]+=dlt[rt];
  31.         sum[RC(rt)]+=(rht[RC(rt)]-lft[RC(rt)]+1)*dlt[rt];
  32.         dlt[rt]=0;
  33.     }
  34. }
  35. void pushup(int rt){
  36.     sum[rt] = sum[LC(rt)] + sum[RC(rt)];
  37. }
  39. void update(int rt,int start,int end,long long int delta)
  40. {
  41.     if(start<=lft[rt]&&end>=rht[rt]){
  42.         dlt[rt]+=delta;
  43.         sum[rt] += (rht[rt]-lft[rt]+1)*delta;
  44.         return;
  45.     }
  46.     pushdown(rt);// 顺便将之前未更新到儿子的信息pushdown
  47.     if(start<=rht[LC(rt)]) update(LC(rt),start,end,delta); //要更新的区间和左儿子有交集(区间左界<=左儿子右界)
  48.     if(end>=lft[RC(rt)]) update(RC(rt),start,end,delta); //要更新的区间和右儿子有交集
  49.     pushup(rt);//返回时候更新当前节点
  50. }
  52.  
  55. void build(int rt,int l,int r)
  56. {
  57.     lft[rt]=l;
  58.     rht[rt]=r;
  59.     dlt[rt]=0;
  60.     if(l==r){
  61.         sum[rt]=init_v[l];
  62.         return;
  63.     }
  64.     else{
  65.         build(LC(rt),l,Mid(l,r));
  66.         build(RC(rt),Mid(l,r)+1,r);
  67.         pushup(rt);
  68.         return;
  69.     } 
  70. }
  72. long long int query(int rt,int start,int end)
  73. {
  74.     if(start>rht[rt]||end<lft[rt]) return 0;
  75. //如果当前节点在查询的范围内就直接返回
  76.     if(start<=lft[rt]&&end>=rht[rt])
  77.         return sum[rt];
  78.     pushdown(rt);//顺便更新之前的标记
  79. // if(start>=lft[RC(rt)]) //区间只和右儿子有交集 区间左界大于等于右儿子左界
  80. // return query(RC(rt),start,end);
  81. // else
  82. // if(end<=rht[LC(rt)]) //区间只和左儿子有集交
  83. // return query(LC(rt),start,end);
  84. // else //左右儿子都有交集
  85.     return query(LC(rt),start,end) + query(RC(rt),start,end);
  86. }
  88.  
  90.  
  93. int main(int argc, char *argv[])
  94. {
  95.     int n,q,i,j,start,end;
  96.     long long int delta;
  97.     char ch[2];
  98.     while(scanf("%d%d",&n,&q)!=EOF)
  99.     {
  100.         for(i=1;i<=n;i++) scanf("%lld",init_v+i);
  101.         build(1,1,n);
  102.         while(q--)
  103.         {
  104.             scanf("%s",ch);
  105.             if(ch[0]=='C')
  106.             {
  107.                 scanf("%d%d%lld",&start,&end,&delta);
  108.                 update(1,start,end,delta);
  109.             }
  110.             else
  111.             {
  112.                 scanf("%d%d",&start,&end);
  113.                 printf("%lld
",query(1,start,end));
  114.             }
  115.         }
  116.         return 0;
  117.     }
  118. }
  120.





附件列表

    【推广】 免费学中医，健康全家人
    原文地址：https://www.cnblogs.com/sober-reflection/p/6c9a73c89008dc456abec64956fee8a3.html