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
太不容易了,无数次超时,超时原因:使用cin输入的,(就算用加速也没用),最后改成了scanf输入才过了。
#include<bits/stdc++.h>
using namespace std;
#define maxn 100007
long long int sum[4*maxn];//每段求和保存
long long int a[maxn],n;//n个节点
void pushup(int rt)//更新节点信息,这里是求和
{
sum[rt]=sum[2*rt]+sum[2*rt+1];
}
void build(int l,int r,int rt)//建立线段树,l,r表示当前节点区间
{//rt表示当前节点
if(l==r)//说明到达叶子节点
{
sum[rt]=a[l];
return;
}
int m=(l+r)>>1;//中点
build(l,m,rt<<1);//左半区间递归
build(m+1,r,rt<<1|1);//右半区间递归
pushup(rt);//更新节点数据
}
void updata(int L,long long int c,int l,int r,int rt)//对a[L]+c;
{
if(l==r)
{
sum[rt]+=c;
return ;
}
int m=(r+l)>>1;
if(L<=m)//判断是左子树还是右子树
updata(L,c,l,m,rt<<1);
else
updata(L,c,m+1,r,rt<<1|1);
pushup(rt);//更新节点信息
}
long long query(int L,int R,int l,int r,int rt)
{//求和操作,[L,R]表示操作区间,[l,r]表示当前区间
if(l>=L&&r<=R)//区间内直接返回
{
return sum[rt];
}
int m=(r+l)>>1;
long long int ans=0;//累加答案
if(L<=m)//若左子区间与[L,R]有重合,递归
ans+=query(L,R,l,m,rt<<1);
if(R>m)
ans+=query(L,R,m+1,r,rt<<1|1);
return ans;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
memset(sum,0,sizeof(sum));//清空sum数组
for(int i=1; i<=n; i++)
scanf("%lld",&a[i]);//数组
build(1,n,1);//建树
int m;
scanf("%d",&m);
while(m--)
{
int t,q;
scanf("%d%d",&t,&q);
if(t==1)
{
long long int v;
scanf("%lld",&v);
updata(q,v,1,n,1);
}
else
{
int v;
scanf("%d",&v);
printf("%lld\n",query(q,v,1,n,1));
}
}
}
return 0;
}