POJ-3468-A Simple Problem with Integers(线段树区间修改+区间求和)

与点修改的区别就是update和query的时候,增加一个add数组,update的时候修改这个add数组。如果树上一点表示的区间在要修改的区间内,那么修改他的add和sum就行,如果这个点表示的区间部分在要修改的区间外,那么将他的add下移,并更新下移区间的sum,这样一次update中最多修改2*h(h为线段树的深度)个结点的add和sum。当左右区间update结束后,修改这个区间的sum为左右两个区间的sum和。query操作同理。

#include
#include
using namespace std;
typedef long long ll;
const int maxn=1e5+7;
ll a[maxn<<2],add[maxn<<2],sum[maxn<<2];
void push_down(int now,int len)
{
    if(!add[now]) return ;
    add[now<<1]+=add[now];
    add[now<<1|1]+=add[now];
    sum[now<<1]+=add[now]*(len-(len>>1));
    sum[now<<1|1]+=add[now]*(len>>1);
    add[now]=0;
}
void push_up(int now)
{
    sum[now]=sum[now<<1]+sum[now<<1|1];
}
void update(int now,int l,int r,int ul,int ur,ll val)
{
    if(ul<=l&&ur>=r)
    {
        add[now]+=val;
        sum[now]+=val*(r-l+1);
        return ;
    }
    if(l==r) return ;
    push_down(now,r-l+1);
    int mid=(l+r)>>1;
    if(ul<=mid) update(now<<1,l,mid,ul,ur,val);
    if(ur>mid)  update(now<<1|1,mid+1,r,ul,ur,val);
    push_up(now);
}
void build(int now,int l,int r)
{
    if(l==r)
    {
        add[now]=0;
        scanf("%I64d",&sum[now]);
        return ;
    }
    int mid=(l+r)>>1;
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    push_up(now);
}
ll query(int now,int l,int r,int ql,int qr)
{
    ll res=0;
    int mid=(l+r)>>1;
    if(ql<=l&&qr>=r) return sum[now];
    push_down(now,r-l+1);
    if(ql<=mid) res+=query(now<<1,l,mid,ql,qr);
    if(qr>mid) res+=query(now<<1|1,mid+1,r,ql,qr);
    return res;
}
int main()
{
    int n,m,l,r,c;
    char option[3];
    while(~scanf("%d%d",&n,&m))
    {
        build(1,1,n);
        for(int i=0;i


转载于:https://www.cnblogs.com/GrowingJlx/p/6642644.html

你可能感兴趣的:(POJ-3468-A Simple Problem with Integers(线段树区间修改+区间求和))