SDUT3771数字计算机

Problem Description

bLue 有一个神器的机器,这个机器可以读入一个数组,并按照用户要求快速地进行数组的处理和计算,它支持如下两种操作:

  • 操作 1:把数组中第 p 个元素的值增加 v。
  • 操作 2:计算数组中 [l, r] 区间内所有数的和。

这个机器就是这么的神奇,但是 bLue 的计算机坏掉了,你能帮他修一下吗?

Input

输入数据有多组(数据组数不超过 20),到 EOF 结束。

对于每组数据:

  • 第 1 行输入一个整数 n (1 <= n <= 10^5),表示数组中元素的个数。
  • 第 2 行输入 n 个用空格隔开的整数 ai (1 <= ai <= 10^10),表示初始输入到计算机中的数组。
  • 第 3 行输入一个整数 q (1 <= q <= 50000),表示用户的操作次数。
  • 接下来 q 行,每行输入先输入 1 个整数,表示操作类型,根据不同的操作类型:
    • 如果类型为 1,则紧接着输入 2 个用空格隔开的整数 p (1 <= p <= n) 和 v (1 <= v <= 10^10),表示要把数组中第 p 个数的值增加 v。
    • 如果类型为 2,则紧接着输入 2 个用空格隔开的整数 l, r (1 <= l <= r <= n),表示要计算区间 [l, r] 内所有数的和(数组下标从 1 开始)。

Output

对于每组数据中的每次类型为 2 的操作,输出 1 行,包含一个整数,表示计算出的和。

Sample Input

5
1 2 3 4 5
5
2 1 2
2 1 5
1 4 10
2 4 5
2 1 5

Sample Output

3
15
19
25
#include
using namespace std;
struct node
{
    int l, r;
    long long value;
};
node tree[1000010];
long long data[1000010];
int n;
void biuld(int root, int l, int r)
{
    tree[root].l = l;
    tree[root].r = r;
    if(l == r)
        tree[root].value = data[l];
    else
    {
        int mid = (l+r)/2;
        biuld(root*2+1, l, mid);
        biuld(root*2+2, mid+1, r);
        tree[root].value = tree[root*2+1].value + tree[root*2+2].value;
    }
}
long long query(int root, int ql ,int qr)
{
    int nl = tree[root].l, nr = tree[root].r;
    if(nl > qr || ql > nr)
        return 0;
    ql = max(nl, ql);
    qr = min(qr, nr);
    if(nr == qr && nl == ql)
        return tree[root].value;
    return query(root*2+1, ql, qr) + query(root*2+2, ql, qr);
}
void update(int root, long long pos, long long value)
{
    int l = tree[root].l, r = tree[root].r;
    if(l > pos || r < pos)
        return ;
    tree[root].value += value;
    if(l == r)
        return ;
    update(root*2+1, pos, value);
    update(root*2+2, pos, value);
}
int main()
{
    while(~scanf("%d", &n))
    {
        memset(tree, 0, sizeof(tree));
        memset(data, 0, sizeof(data));
        for(int i = 0; i < n; i++)
            scanf("%lld", &data[i]);
        biuld(0, 0, n-1);
        int q;
        scanf("%d", &q);
        while(q--)
        {
            long long x, a, b;
            scanf("%lld%lld%lld", &x, &a, &b);
            if(x==2)
                printf("%lld\n", query(0, a-1, b-1));
            if(x==1)
                update(0, a-1, b);
        }
    }
}

你可能感兴趣的:(线段树)