题目链接:http://poj.org/problem?id=3468
思路:线段树实现部分区域修改,区域求和。
进行区域修改时,如果区域可以覆盖该节点所代表的区域,用lnc记录每个值需要增加的量,并且不在下传;如果不能覆盖则用nsum记录这个节点需要加的值,继续下传。
进行查询时如果查询的点覆盖这个节点,ans+=nsum+lnc*区间大小,如果不能覆盖,则将该节点lnc传到该节点的nsum中,并将lnc下传到其子节点
注意:注意__int64计数;
注意R、L与r、l的区分;
#include<cstdio>
using namespace std;
const int num=100005;
int n,m;
__int64 ans;
struct node
{
int l,r;
__int64 nsum,lnc;
}data[num<<2];
void build(int rt,int l,int r)
{
data[rt].l=l;
data[rt].r=r;
data[rt].nsum=0;
data[rt].lnc=0;
int mid=(l+r)>>1;
if(l!=r)
{
build(rt*2,l,mid);
build(rt*2+1,mid+1,r);
}
}
void Insert(int rt,int L,int R,int a)
{
if(data[rt].l==L&&data[rt].r==R)
{
data[rt].lnc+=a;
return ;
}
data[rt].nsum+=(R-L+1)*a;
int mid=(data[rt].l+data[rt].r)>>1;
if(R<=mid)
Insert(rt*2,L,R,a);
else if(L>mid)
Insert(rt*2+1,L,R,a);
else
{
Insert(rt*2,L,mid,a);
Insert(rt*2+1,mid+1,R,a);
}
}
void query(int rt,int L,int R)
{
if(data[rt].l==L&&data[rt].r==R)
{
ans+=(data[rt].nsum+data[rt].lnc*(R-L+1));
return ;
}
if(data[rt].lnc!=0)
{
data[rt].nsum+=(data[rt].r-data[rt].l+1)*data[rt].lnc;
data[rt*2].lnc+=data[rt].lnc;
data[rt*2+1].lnc+=data[rt].lnc;
data[rt].lnc=0;
}
int mid=(data[rt].l+data[rt].r)>>1;
if(R<=mid)
query(rt*2,L,R);
else if(L>mid)
query(rt*2+1,L,R);
else
{
query(rt*2,L,mid);
query(rt*2+1,mid+1,R);
}
}
int main()
{
int i,a,b,c;
char s[3];
//freopen("in.txt","r",stdin);
scanf("%d%d",&n,&m);
build(1,1,n);
for(i=1;i<=n;i++)
{
scanf("%d",&a);
Insert(1,i,i,a);
}
for(i=0;i<m;i++)
{
scanf("%s",s);
if(s[0]=='Q')
{
scanf("%d%d",&a,&b);
ans=0;
query(1,a,b);
printf("%I64d\n",ans);
}
else
{
scanf("%d%d%d",&a,&b,&c);
Insert(1,a,b,c);
}
}
return 0;
}