码迷,mamicode.com
首页 > 其他好文 > 详细

POJ - 3468 A Simple Problem with Integers(线段树区间更新)

时间:2015-11-11 20:44:14      阅读:185      评论:0      收藏:0      [点我收藏+]

标签:

1、给出了一个序列,你需要处理如下两种询问。

"C a b c"表示给[a, b]区间中的值全部增加c (-10000 ≤ c ≤ 10000)。

"Q a b" 询问[a, b]区间中所有值的和。

2、线段树单点更新太费时,所以使用区间更新

3、

技术分享
#include <cstdio>

#define L(root) ((root) << 1)
#define R(root) (((root) << 1) + 1)

const int MAXN = 100001;
int numbers[MAXN];

struct st
{
    // 区间范围
    int left, right;
    // 更新值、区间总和
    long long delta, sum;
} st[MAXN * 4];

// 建树代码基本不变
void build(int root, int l, int r)
{
    st[root].left = l, st[root].right = r, st[root].delta = 0;
    if (l == r)
    {
        st[root].sum = numbers[l];
        return;
    }

    int m = l + ((r - l) >> 1);
    build(L(root), l, m);
    build(R(root), m + 1, r);
    st[root].sum = st[L(root)].sum + st[R(root)].sum;
}

long long query(int root, int l, int r)
{
    // 如查询区间恰等于节点区间,直接返回该区间总和即可
    if (st[root].left == l && st[root].right == r)
    {
        return st[root].sum;
    }

    // 否则需将当前区间的“缓冲”值更新下去并修正各节点区间的总和
    if (st[root].delta)
    {
        st[L(root)].delta += st[root].delta;
        st[R(root)].delta += st[root].delta;
        st[L(root)].sum += st[root].delta * (st[L(root)].right - st[L(root)].left + 1);
        st[R(root)].sum += st[root].delta * (st[R(root)].right - st[R(root)].left + 1);
        st[root].delta = 0;
    }

    int m = st[root].left + ((st[root].right - st[root].left) >> 1);
    if (r <= m)
    {
        return query(L(root), l, r);
    }
    else if (l > m)
    {
        return query(R(root), l, r);
    }
    else
    {
        return query(L(root), l, m) + query(R(root), m + 1, r);
    }
}

void update(int root, int l, int r, long long v)
{
    // 如变更区间恰等于节点区间,只修正当前节点区间即可
    if (st[root].left == l && st[root].right == r)
    {
        st[root].delta += v;
        st[root].sum += v * (r - l + 1);
        return;
    }

    // 否则需向下修正相关节点区间
    if (st[root].delta)
    {
        st[L(root)].delta += st[root].delta;
        st[R(root)].delta += st[root].delta;
        st[L(root)].sum += st[root].delta * (st[L(root)].right - st[L(root)].left + 1);
        st[R(root)].sum += st[root].delta * (st[R(root)].right - st[R(root)].left + 1);
        st[root].delta = 0;
    }

    int m = st[root].left + ((st[root].right - st[root].left) >> 1);
    if (r <= m)
    {
        update(L(root), l, r, v);
    }
    else if (l > m)
    {
        update(R(root), l, r, v);
    }
    else
    {
        update(L(root), l, m, v);
        update(R(root), m + 1, r, v);
    }
    // 同时一定要记得修正当前节点区间的总和
    st[root].sum = st[L(root)].sum + st[R(root)].sum;
}

int main()
{
    int N, Q;
    while (scanf("%d%d", &N, &Q) != EOF)
    {
        for (int i = 1; i <= N; ++i)
        {
            scanf("%d", &numbers[i]);
        }

        build(1, 1, N);

        char cmd;
        int l, r;
        long long v;
        while (Q--)
        {
            scanf(" %c", &cmd);
            scanf("%d%d", &l, &r);
            switch (cmd)
            {
            case Q:
                printf("%lld\n", query(1, l, r));
                break;

            case C:
                scanf("%lld", &v);
                if (v)
                {
                    update(1, l, r, v);
                }
                break;
            }
        }
    }

    return 0;
}
View Code

 

POJ - 3468 A Simple Problem with Integers(线段树区间更新)

标签:

原文地址:http://www.cnblogs.com/bofengyu/p/4957239.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!