数组计算机

p.s. 线段树基本操作（点修改）

数组计算机

Time Limit: 1000 ms Memory Limit: 65536 KiB

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

题目链接：

http://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Contest/contestproblem/cid/2715/pid/3771

#include <bits/stdc++.h>

using namespace std;

long long sum[400001];
long long num[100001];

void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=num[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void update(int l,int r,int L,long long x,int rt)
{
    if(l==r)
    {
        sum[rt]+=x;
        return;
    }
    int mid=(l+r)>>1;
    if(L<=mid)
    {
        update(l,mid,L,x,rt<<1);
    }
    else
    {
        update(mid+1,r,L,x,rt<<1|1);
    }
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

long long query(int l,int r,int L,int R,int rt)//单点修改
{
    if(L<=l&&R>=r)
    {
        return sum[rt];
    }
    int mid=(r+l)>>1;
    long long ans=0;
    if(L<=mid)
    {
        ans+=query(l,mid,L,R,rt<<1);
    }
    if(R>mid)
    {
        ans+=query(mid+1,r,L,R,rt<<1|1);
    }
    return ans;
}

int main()
{
    int n;
    while(cin >> n)
    {
        memset(sum,0,sizeof(sum));
        memset(num,0,sizeof(num));
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&num[i]);
        }
        build(1,n,1);
        int q;
        cin >> q;
        while(q--)
        {
            int a;
            scanf("%d",&a);
            if(a==1)
            {
                int p;
                long long v;
                scanf("%d%lld",&p,&v);
                update(1,n,p,v,1);
            }
            if(a==2)
            {
                int l,r;
                scanf("%d%d",&l,&r);
                cout << query(1,n,l,r,1) << endl;
            }
        }
    }
    return 0;
}