POJ 3468 A Simple Problem with Integers
传送门
题目大意:
给你一个序列,和一些操作
操作分为
C A B C
意思是将区间[A,B]的值加上C
Q A B
意思是查询[A,B]区间的和
分析:
线段树区间更新
加一个add的懒惰标记
同时维护一个sum
注意
可能爆int
Tips:超时可以试试C++,我第一次写到丑,G++超时,C++ AC了。
这里主要还是理解自上而下 自下而上 八个字
代码如下:
#include<cstdio>
#define ll long long
const int Max=100005;
struct node
{
int lson,rson;
ll sum,add;
}tree[Max<<2];
int n,q,a,b,c;
char s[5];
void build(int l,int r,int n)
{
tree[n].lson=l;
tree[n].rson=r;
if(l==r)
{
scanf("%lld",&tree[n].sum);
return;
}
int mid=(l+r)>>1;
build(l,mid,n<<1);
build(mid+1,r,n<<1|1);
tree[n].sum=tree[n<<1].sum+tree[n<<1|1].sum;
}
void update(int l,int r,int v,int n)
{
if(tree[n].lson==l&&tree[n].rson==r)
{
tree[n].add+=v;
return;
}
tree[n].sum+=(r-l+1)*(ll)v;
//注意是 r-l+1
int mid=(tree[n].lson+tree[n].rson)>>1;
if(l>mid)update(l,r,v,n<<1|1);
else if(r<=mid)update(l,r,v,n<<1);
else update(l,mid,v,n<<1),update(mid+1,r,v,n<<1|1);
}
ll query(int l,int r,int n)
{
if(tree[n].lson==l&&tree[n].rson==r)
return tree[n].sum+(tree[n].rson-tree[n].lson+1)*tree[n].add;
if(tree[n].add)//传递add
{
tree[n<<1].add+=tree[n].add;
tree[n<<1|1].add+=tree[n].add;
tree[n].sum+=(tree[n].rson-tree[n].lson+1)*tree[n].add;
//传递到时候顺便更新sum,免得一会再算
tree[n].add=0;
}
int mid=(tree[n].lson+tree[n].rson)>>1;
if(l>mid)return query(l,r,n<<1|1);
else if(r<=mid)return query(l,r,n<<1);
else return query(l,mid,n<<1)+query(mid+1,r,n<<1|1);
}
int main()
{
scanf("%d%d",&n,&q);
build(1,n,1);
while(q--)
{
scanf("%s%d%d",s,&a,&b);
if(s[0]=='C')scanf("%d",&c),update(a,b,c,1);
else printf("%lld\n",query(a,b,1));
}
return 0;
}