数组计算机(讲课)
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
Output
3
15
19
25
#include <iostream>
#include <bits/stdc++.h>
//#define ll long long
#define maxn 100005
using namespace std;
long long a[maxn], tree[maxn*4];
void Build(long long node, long long left, long long right)//建树;
{
if(left == right)
{
tree[node] = a[left];
return ;//返回空表示线段的左右端点已经相等;
}
int mid = (left + right)/2;
Build(node*2, left, mid);
Build(node*2+1, mid+1, right);
tree[node] = tree[node*2] + tree[node*2+1];
}
void updata(long long node, long long left, long long right, long long x, long long r)//节点更新;
{
if(left == right)
{
tree[node] += r;
return ;//返回空代表已经找到要更新的节点;
}
int mid = (left+right)/2;
if(x <= mid)
{
updata(node*2, left, mid, x, r);
}
else
{
updata(node*2+1, mid+1, right, x, r);
}
tree[node] = tree[node*2] + tree[node*2+1];
}
long long Find(long long node, long long left, long long right, long long x, long long r)//区间查询;
{
long long res = 0;
if(x <= left && r >= right)
{
return tree[node];
}
int mid = (left+right)/2;
if(x <= mid)
{
res += Find(node*2, left, mid, x, r);
}
if(r > mid)
{
res += Find(node*2+1, mid+1, right, x, r);
}
return res;
}
int main()
{
int n, i;
while(~scanf("%d", &n))
{
memset(a, 0, sizeof(a));
memset(tree, 0, sizeof(tree));
for(i = 1; i <= n; i++)
{
scanf("%lld", &a[i]);
}
Build(1, 1, n);
int q;
//for(i = 0; i < n; i++)
//{
// printf("%lld ", tree[i]);
//}
//printf("\n");
scanf("%d", &q);
while(q--)
{
long long p, x, r;
scanf("%lld %lld %lld", &p, &x, &r);
if(p == 1)
{
updata(1, 1, n, x, r);
//for(i = 0; i < n; i++)
//{
//printf("%lld ", tree[i]);
// }
//printf("\n");
}
else if(p == 2)
{
long long h = Find(1, 1, n, x, r);
printf("%lld\n", h);
}
}
}
return 0;
}