poj - 4368

poj - 3468 线段树区间修改

这题是个模板题， 很有代表意义，poj的数据确实有点庞大，tle 好久，最后还是两秒不到就过了。今天初学线段树区间修改，写个博客纪念一下，代码区间修改的代码还没有成熟，慢慢就好了，直接上码：

#pragma G++ optimize(2)

#pragma GCC optimize(2)

#include<iostream>

#include<algorithm>

#include<stack>

#include<vector>

#include<iostream>

#include<climits>

#include<queue>

#include<cassert>

#include<iomanip>

#include<cmath>

#include<string> 

#include<cstdio>

#include<cstring>

#define _rep(i, a, b) for(int i = (a); i <=
(b); ++i)

#define _rev(i, a, b) for(int i = (a); i >=
(b); --i)

#define _for(i, a, b) for(int i = (a); i <(b);
++i)

#define _rof(i, a, b) for(int i = (a); i >(b);
--i)

#define maxn 100009

#define maxm 109

#define ll long long

#define met(a,b)
memset((a),(b), sizeof(a))

#define db double

#define oo INT_MAX

#define eps 1e-8

using namespace std;

int n, m,
a[maxn];

struct node

{

    ll l, r;

    ll sum, add;

#define lson t[o << 1]

#define rson t[o << 1 | 1]

#define le  o << 1

#define ri  o << 1 | 1

}t[maxn * 4];

void update(int o) {

    t[o].sum = lson.sum + rson.sum;

}

void build(int o, int l, int r) {

    t[o].l = l, t[o].r = r;

    if (l == r) { t[o].sum = a[l]; return; }

    int mid = (l + r) >> 1;

    build(le, l, mid);

    build(ri, mid + 1, r);

    update(o);

}

 

void spread(int o) {

    if (t[o].add) {

         lson.sum += t[o].add * (lson.r - lson.l + 1);

         rson.sum += t[o].add * (rson.r - rson.l + 1);

         lson.add += t[o].add;

         rson.add += t[o].add;

         t[o].add = 0;

    }

}

 

void change(int o, int l, int r, int d) {

    if (l <= t[o].l && t[o].r <= r) {

         t[o].sum += (ll)d * (t[o].r - t[o].l + 1);

         t[o].add += d;

         return;

    }

    spread(o);

    int mid = (t[o].l + t[o].r) >> 1;

    if (mid >= l) change(le, l, r, d);

    if (r > mid)change(ri, l, r, d);

    update(o);

}

 

ll ask(int o, int l, int r) {

    if (l <= t[o].l && r >= t[o].r)return t[o].sum;

    spread(o);

    int mid = (t[o].l + t[o].r ) >> 1;

    ll ret = 0;

    if (l <= mid)ret +=
ask(le, l, r);

    if (r > mid) ret +=
ask(ri, l, r);

    return ret;

}

 

int main()
{

    scanf("%d%d", &n, &m);

    _rep(i, 1, n) {

         scanf("%d", &a[i]);

    }

    build(1,
1, n);

    char op[2];

    _rep(i, 1, m) {

         scanf(" %s", &op);

         int l, r, d;

         if (op[0] == 'Q') {

             scanf("%d%d", &l, &r);

             printf("%lld\n", ask(1,
l, r).sum);

         }

         else {

             scanf("%d%d%d",
&l, &r, &d);

             change(1,
l, r, d);

         }

    }

}