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;
}