线段树模板

POJ - 3468
https://cn.vjudge.net/problem/POJ-3468

#include<bits/stdc++.h> 
using namespace std;
typedef long long ll;

const ll maxn = 4 * 1e5 + 500;
ll n,m;

struct SegTree
{
    ll l;
    ll r;
    ll sum;
    ll add;
};

SegTree segTree[maxn];
char c;
ll l,r,v;

void PushUp(ll i)
{
    segTree[i].sum = segTree[i<<1].sum + segTree[i<<1|1].sum;
}

void PushDown(ll i)
{
    if(segTree[i].add){
        ll add = segTree[i].add;
        ll len = segTree[i].r - segTree[i].l + 1;
        segTree[i<<1].add += add;
        segTree[i<<1|1].add += add;
        segTree[i<<1|1].sum += len/2 * add;
        segTree[i<<1].sum += (len-len/2) * add;
        segTree[i].add = 0;
    }
}

void Build(ll l, ll r, ll i)
{
    segTree[i].l = l;
    segTree[i].r = r;
    segTree[i].add = 0;
    if(l == r){cin>>segTree[i].sum; return;}
    ll mid = l + r >> 1;
    Build(l,mid,i<<1);
    Build(mid+1,r,i<<1|1);
    PushUp(i);
}

void Update(ll v, ll l, ll r, ll i)
{
    if(l <= segTree[i].l && segTree[i].r <= r){
        segTree[i].sum += v * (segTree[i].r - segTree[i].l + 1);
        segTree[i].add += v;
        return ;
    }
    PushDown(i);
    ll mid = segTree[i].l + segTree[i].r >> 1;
    ll ret = 0;
    if(l > mid){
        Update(v,l,r,i<<1|1);
    }
    else if(r <= mid){
        Update(v,l,r,i<<1);
    }
    else{
        Update(v,l,mid,i<<1);
        Update(v,mid+1,r,i<<1|1);
    }
    PushUp(i);
    return;
}

ll Query(ll l, ll r, ll i)
{
    if(l <= segTree[i].l && segTree[i].r <= r){
        return segTree[i].sum;
    }
    PushDown(i);
    ll mid = segTree[i].l + segTree[i].r >> 1;
    ll ret = 0;
    if(l > mid){
        ret += Query(l,r,i<<1|1);
    }
    else if(r <= mid){
        ret += Query(l,r,i<<1);
    }
    else{
        ret += Query(l,mid,i<<1);
        ret += Query(mid+1,r,i<<1|1);
    }
    PushUp(i);
    return ret;
}


int main()
{
    //freopen("data.c", "r", stdin);
    ios::sync_with_stdio(0);
    cin.tie();
    cout.tie();
    while(cin>>n>>m){
        Build(1,n,1);
        while(m--){
            cin>>c>>l>>r;
            if(c == 'C') cin>>v,Update(v,l,r,1);
            else if(c == 'Q') cout<<Query(l,r,1)<<endl;
        }
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值