POJ 3468 A Simple Problem with Integers

来源:互联网 发布:移动网络不稳定 编辑:程序博客网 时间:2024/05/02 02:05

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 longconst 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;}
0 0
原创粉丝点击