poj 3468 分类： poj templates 2015-04-01 20:20 34人阅读 评论(0) 收藏

#include<map>
#include<string>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<iostream>
#include<algorithm>
const int MAXN = 100005;

int n , q, a[MAXN] = {0};
struct node
{
long long sum;
long long add;
}tree[MAXN<<2] = {0};

#define L(x) (x<<1)
#define R(x) ((x<<1)|1)

inline void update(int x){tree[x].sum = tree[L(x)].sum + tree[R(x)].sum;}

inline void build(int ll,int rr,int si)
{
if(ll == rr)  tree[si].sum = a[ll];
else
{
int mid = (ll+rr)>>1;
build(ll,mid,L(si));
build(mid+1,rr,R(si));
update(si);
}
}
inline void pushdown(int x,int ll,int rr)
{
int mid = (ll+rr)>>1;

if(tree[x].add)
{
tree[L(x)].sum += tree[x].add*(mid-ll+1);
tree[R(x)].sum += tree[x].add*(rr - mid);
tree[L(x)].add += tree[x].add;
tree[R(x)].add += tree[x].add;
tree[x].add = 0;
}
}
inline void add(int l,int r,int ll,int rr,int si,int val)
{
if(ll == l && rr == r)
tree[si].sum += (long long)val*(rr-ll+1), tree[si].add += val;
else
{
int mid = (ll+rr)>>1;

pushdown(si,ll,rr);
if(r<=mid) add(l,r,ll,mid,L(si),val);
else if(l>mid)add(l,r,mid+1,rr,R(si),val);
else
{
add(l,mid,ll,mid,L(si),val);
add(mid+1,r,mid+1,rr,R(si),val);
}
update(si);
}
}
inline long long count(int l,int r,int ll,int rr,int si)
{
if(ll == l && rr == r)return tree[si].sum;
else
{
int mid = (ll+rr)>>1;

pushdown(si,ll,rr);
if(r<=mid)    return count(l,r,ll,mid,L(si));
else if(l>mid)return count(l,r,mid+1,rr,R(si));
else      return count(l,mid,ll,mid,L(si)) + count(mid+1,r,mid+1,rr,R(si));
update(si);
}
}
#undef L
#undef R
int main()
{
#ifndef ONLINE_JUDGE
freopen("poj3468.in","r",stdin);
freopen("poj3468.out","w",stdout);
#endif

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

for(int i = 1 ; i <= n ;i++)
scanf("%d",&a[i]);

build(1,n,1);

for(int i = 1; i <= q; i++)
{
static int a,b,c;
static char ch[5] = {'\0'};
scanf("%s",ch);scanf("%d%d",&a,&b);

switch(ch[0])
{
case 'C': scanf("%d",&c); add(a,b,1,n,1,c);   break;
case 'Q': printf("%I64d\n",count(a,b,1,n,1)); break;
}
}

#ifndef ONLINE_JUDGE
fclose(stdin);
fclose(stdout);
#endif
return 0;
}