题目链接:http://poj.org/problem?id=3468
题目大意:成段更新,区间求和
解题思路:线段树的基本应用,注意每次更新操作不需要更行到底,只需要标记一下,等下次更新再向下更新。。。
代码如下:
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define LL long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define N 100005
LL sum[N<<2],mark[N<<2];
void pushup(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int m)
{
if(mark[rt])
{
mark[rt<<1]+=mark[rt];
mark[rt<<1|1]+=mark[rt];
sum[rt<<1]+=(m-(m>>1))*mark[rt];
sum[rt<<1|1]+=(m>>1)*mark[rt];
mark[rt]=0;
}
}
void build(int l,int r,int rt)
{
mark[rt]=0;
if(l==r)
{
scanf("%lld",&sum[rt]);
return;
}
int m=(l+r)>>1;
build(lson);
build(rson);
pushup(rt);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if(L<=l&&R>=r)
{
mark[rt]+=c;
sum[rt]+=(r-l+1)*c;
return;
}
pushdown(rt,r-l+1);
int m=(l+r)>>1;
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 sum[rt];
pushdown(rt,r-l+1);
int m=(l+r)>>1;
LL ret=0;
if(L<=m) ret+=query(L,R,lson);
if(R>m) ret+=query(L,R,rson);
return ret;
}
int main()
{
int n,m;
char c[2];
while(~scanf("%d%d",&n,&m))
{
build(1,n,1);
int x,y,a;
for(int i=0;i<m;i++)
{
scanf("%s",c);
if(c[0]=='C')
{
scanf("%d%d%d",&x,&y,&a);
update(x,y,a,1,n,1);
}
else
{
scanf("%d%d",&x,&y);
printf("%lld\n",query(x,y,1,n,1));
}
}
}
return 0;
}