poj 3468 A Simple Problem with Integers || 线段树

poj 3468 A Simple Problem with Integers 请戳

1. 题意：
   给 n 个数，对这组数用两种方法蹂躏：
   * 一种蹂躏方法是给一段区间加上或者减去一个数；
   * 一种是给出一段区间的和；
   好好玩的样子。。。蹂躏起来

2. 思路：
   很明显线段数蹂躏大法好，区间修改与查询。小傻子又写了一个非主流版本，虽然ac了，但是时间还是比和上次的非主流版本一样的时间的。不甘心的我从网上找了标准代码交上去，实现的时间居然和我的是一样的。。。然后看看比我用时短1秒的都是用c++交的，而我一直用g++教。。。（太傻了）

   ps：这次小傻子用了pushdown（），但是写的不好，人又搓（小傻子不是乱叫的）debug，然后和前面的非主流写法对拍了好久才发现bug，诶诶。

3. 复杂度：
   时间复杂度：$O(n) = n \log n$
   空间复杂度： $O(n) = 8 n$

4. 代码：

   1. 非主流写法1

   2. 非主流写法2

/* ***********************************************
Author        :Ilovezilian
Created Time  :2015/9/5 0:11:25
File Name     :seg_tree_5.cpp
************************************************ */

//#include <bits/stdc++.h>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define INF 0x7fffffff
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N = 1e5 + 10, mod = 1e9+7;
typedef long long LL;
LL po[N<<2], add[N<<2];

void pushup(int l, int r, int rt)
{
    po[rt] = po[rt<<1] + po[rt<<1|1] + add[rt] * (r - l + 1);
}

void build(int l, int r, int rt)
{
    add[rt] = 0;
    if(l == r)
    {
        scanf("%I64d", po + rt);
        return;
    }

    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    pushup(l, r, rt);
}

void update(int val, int L, int R, int l, int r, int rt)
{
    if(L <= l && r <= R)
    {
        add[rt] += val;
        po[rt] += (r - l + 1) * val;
        return;
    }

    int m = (l + r) >> 1;
    if(L <= m) update(val, L, R, lson);
    if(R > m) update(val, L, R, rson);
    pushup(l, r, rt);
}

LL query(int L, int R, int l, int r, int rt)
{
    if(L <= l && r <= R)
    {
        //printf("po(%d,%d) = %I64d\n", l, r, po[rt]);
        return po[rt];
    }

    int m = (l + r) >> 1;
    LL ret = 0;
    if(L <= m) ret += query(L, R, lson) + add[rt] * (min(m - max(l, L) + 1, R - max(l, L) + 1));
    if(R > m) ret += query(L, R, rson) + add[rt] * (min(min(R, r) - m, min(R, r) - L + 1));
    //printf("po(%d,%d) = %I64d ret = %I64d\n", l, r, po[rt], ret);
    return ret;
}

void solve()
{
    int n, q;
    while(~scanf("%d%d", &n, &q))
    {
        build(1, n, 1);

        char s[3];
        int a, b, c;
        for(int i = 0; i < q; i ++)
        {
            scanf("%s", s);
            if(s[0] == 'C')
            {
                scanf("%d%d%d", &a, &b, &c);
                update(c, a, b, 1, n, 1);
            }
            else
            {
                scanf("%d%d", &a, &b);
                printf("%I64d\n", query(a, b, 1, n, 1)); 
            }
        }
    }
}

int main()
{
    //freopen("test.out","r",stdin);
    //freopen("ac.out","w",stdout);
    solve();
    return 0;
}

3. 主流标准写法

#include<cstdio>
#include<algorithm>
using namespace std;

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long 
const int maxn = 111111;
LL add[maxn<<2], sum[maxn<<2];

void pushup(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

void pushdown(int rt, int m)
{
    if(add[rt])
    {
        add[rt<<1] += add[rt];
        add[rt<<1|1] += add[rt];
        sum[rt<<1] += add[rt] * (m - (m >> 1));
        sum[rt<<1|1] += add[rt] * (m >> 1);
        add[rt] = 0;
    }
}

void build(int l, int r, int rt)
{
    add[rt] = 0;
    if(l == r)
    {
        scanf("%I64d", sum + rt);
        return;
    }

    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update(int L, int R, int c, int l, int r, int rt)
{
    if(L <= l && r <= R)
    {
        add[rt] += c;
        sum[rt] += c * (r - l + 1);
        return;
    }

    pushdown(rt, r - l + 1);
    int m = (l + r) >> 1;
    if(L <= m) update(L, R, c, lson);
    if(R > m) update(L, R, c, rson);
    pushup(rt);
}

LL query(int L, int R, int l, int r, int rt)
{
    if(L <= l && r <= R)
    {
        return sum[rt];
    }
    pushdown(rt, r - l + 1);
    int m = (l + r) >> 1;
    LL ret = 0;
    if(L <= m) ret += query(L, R, lson);
    if(R > m) ret += query(L, R, rson);
    return ret;
}

int main()
{
    int N, Q;
    scanf("%d%d", &N, &Q);
    build(1, N, 1);
    while( Q --)
    {
        char op[2];
        int a, b, c;
        scanf("%s", op);
        if(op[0] == 'Q')
        {
            scanf("%d%d", &a, &b);
            printf("%I64d\n", query(a, b, 1, N, 1));
        }
        else
        {
            scanf("%d%d%d", &a, &b, &c);
            update(a, b, c, 1, N, 1);
        }
    }
    return 0;
}