POJ 3468.A Simple Problem with Integers-线段树(成段增减、区间查询求和)

POJ 3468.A Simple Problem with Integers

这个题就是成段的增减以及区间查询求和操作。

代码:

 #include<iostream>

 #include<cstdio>

 #include<cstring>

 #include<cmath>

 #include<cstdlib>

 #include<algorithm>

 #include<queue>

 #include<map>

 using namespace std;

 typedef long long ll;

 const int maxn=1e5+;

 const int inf=0x3f3f3f3f;

 #define lson l,m,rt<<1

 #define rson m+1,r,rt<<1|1

 ll tree[maxn<<],add[maxn<<];

 void pushup(int rt)

 {

     tree[rt]=tree[rt<<]+tree[rt<<|];

 }

 void pushdown(int rt,int m)

 {

     if(add[rt]){

         add[rt<<]+=add[rt];

         add[rt<<|]+=add[rt];

         tree[rt<<]+=(m-(m>>))*add[rt];

         tree[rt<<|]+=(m>>)*add[rt];

         add[rt]=;

     }

 }

 void build(int l,int r,int rt)

 {

     add[rt]=;

     if(l==r){

         scanf("%lld",&tree[rt]);

         return ;

     }

     int m=(l+r)>>;

     build(lson);

     build(rson);

     pushup(rt);

 }

 void update(int L,int R,ll c,int l,int r,int rt)

 {

     if(L<=l&&r<=R){

         add[rt]+=c;

         tree[rt]+=c*(r-l+);

         return ;

     }

     pushdown(rt,r-l+);

     int m=(l+r)>>;

     if(L<=m) update(L,R,c,lson);

     if(R> m) update(L,R,c,rson);

     pushup(rt);

 }

 ll query(int L,int R,int l,int r,int rt)

 {

     if(L<=l&&r<=R){

         return tree[rt];

     }

     pushdown(rt,r-l+);

     int m=(l+r)>>;

     ll ret=;

     if(L<=m)ret+=query(L,R,lson);

     if(R> m)ret+=query(L,R,rson);

     return ret;

 }

 int main()

 {

     int n,m;

     scanf("%d%d",&n,&m);

     build(,n,);

     char s[];int l,r;ll c;

     for(int i=;i<m;i++){

         scanf("%s",s);

         if(s[]=='Q'){

             scanf("%d%d",&l,&r);

             printf("%lld\n",query(l,r,,n,));

         }

         else{

             scanf("%d%d%lld",&l,&r,&c);

             update(l,r,c,,n,);

         }

     }

     return ;

 }