POJ3468 A Simple Problem with Integers 线段树成段更新(lazy)

题目连接:http://poj.org/problem?id=3468


题目大意:给出一个100000的序列,进行100000次操作,操作"C l r v"表示将区间[l,r]内的所有元素的值加上v;操作"Q l r"表示查询区间[l,r]的元素的和。


实现代码如下:

//线段树成段更新:每次更新一个区间[l,r]
#include <cstdio>
using namespace std;
const int maxn=100005;
typedef long long ll;
ll num[maxn]; //纪录每个数的初始值
struct segment
{
    int l,r;
    int mid()
    {
        return (l+r)>>1;
    }
    ll sum; //纪录该区间原来的和
    ll add; //纪录区间的增量
}tree[maxn<<2]; //区间和为sum+add*(r-l+1);
void build(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].add=0;
    if(l==r)
    {
        tree[root].sum=num[l];
        return ;
    }
    int mid=tree[root].mid();
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
    tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;
}
void update(int root,int l,int r,int v)
{
    if(tree[root].l==l&&tree[root].r==r)
    {
        tree[root].add+=v;
        return ;
    }
    tree[root].sum+=(ll)v*(r-l+1);
    int mid=tree[root].mid();
    if(r<=mid) update(root<<1,l,r,v);
    else if(l>mid) update(root<<1|1,l,r,v);
    else
    {
        update(root<<1,l,mid,v);
        update(root<<1|1,mid+1,r,v);
    }
}
ll query(int root,int l,int r)
{
    if(tree[root].l==l&&tree[root].r==r)
      return tree[root].sum+tree[root].add*(r-l+1);
    tree[root].sum+=tree[root].add*(tree[root].r-tree[root].l+1);
    int mid=tree[root].mid();
    update(root<<1,tree[root].l,mid,tree[root].add);
    update(root<<1|1,mid+1,tree[root].r,tree[root].add);
    tree[root].add=0;
    if(r<=mid) return query(root<<1,l,r);
    else if(l>mid) return query(root<<1|1,l,r);
    else return query(root<<1,l,mid)+query(root<<1|1,mid+1,r);
}
int main()
{
    int n,m;
    char ch[5];
    while(scanf("%d%d",&n,&m)!=-1)
    {
        for(int i=1;i<=n;i++)
          scanf("%lld",&num[i]);
        build(1,1,n);
        while(m--)
        {
            int l,r,v;
            scanf("%s",ch);
            if(ch[0]=='C') //更新区间[l,r]
            {
                scanf("%d%d%d",&l,&r,&v);
                update(1,l,r,v);
            }
            else //查询区间[l,r]
            {
                scanf("%d%d",&l,&r);
                printf("%lld\n",query(1,l,r));
            }
        }
    }
    return 0;
}


你可能感兴趣的:(POJ3468 A Simple Problem with Integers 线段树成段更新(lazy))