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 ;
 }