3198: 区间和
时间限制(普通/Java):1000MS/3000MS 内存限制:65536KByte
总提交: 967 测试通过:302
描述
给定n个数据,有两个操作,加减其中的一个数据,当然还可查询在某段数据的和。
输入
输入数据有多组,每组数据的
第一行输入n,1=<n<=500000,代表数据的个数。
第二行输入具体数据,数据为正整数,范围在1到10000.
第三行输入m,1<=m<=100000,表示操作的次数。包含了修改和查询操作。
下面m行就是具体的操作了。
C i x 表示为第i个元素加上x,x范围在1到10000.
Q i j 表示查询区段i到j的和。保证输入的i<=j.
以EOF结束。
输出
输出查询后的区段和。
样例输入
8
1 5 9 11 2 8 15 6
4
Q 1 3
C 2 10
Q 1 4
Q 2 5
样例输出
15
36
37
提示:类型最好定义为__int64
题解:~~~~~~~线段数,单点修改和区间查询,父节点记录和就好了,用一个数组去存树,一般开4倍
#include<cstdio>
#include<iostream>
#include<string.h>
using namespace std;
const int maxn=500005;
int arr[maxn],tree[maxn*4];
void build(int node,int l,int r){ //建树
if(l==r)
tree[node] = arr[l]; // 走到最深的节点,赋值
else{
int mid = (l+r)/2;
int left_node = 2*node + 1; //左儿子
int right_node = 2*node + 2; //右儿子
build(left_node,l,mid);
build(right_node,mid+1,r); // 两边递归
tree[node]=tree[left_node]+tree[right_node]; // 回溯父节点记录和
}
}
void update(int node,int l,int r,int idx,int val){
if(l==r)
arr[idx]+=val,tree[node]+=val;//找到点
else{
int mid = (l+r)/2;
int left_node = 2*node + 1;
int right_node = 2*node + 2;
if(idx<=mid&&idx>=l) //比中点小 左边搜
update(left_node,l,mid,idx,val);
else
update(right_node,mid+1,r,idx,val);
tree[node]=tree[left_node]+tree[right_node]; //修改完回溯
} //修改父节点的和
}
int query(int node,int l,int r,int L,int R){
if(r<L || l>R) //越界不走,优化
return 0;
else if ((l>=L && r<=R) || l==r) //区间内直接返回
return tree[node];
else{
int mid = (l+r)/2;
int left_node = 2*node + 1;
int right_node = 2*node + 2;
int query_left=query(left_node,l,mid,L,R); //左边搜
int query_right=query(right_node,mid+1,r,L,R);//右边搜
return query_left+query_right; //返回和
}
}
int main(){
int n;
while(~scanf("%d",&n)){
memset(tree,0,sizeof tree);
for(int i=0;i<n;i++)
scanf("%d",&arr[i]);
build(0,0,n-1);
int m;
scanf("%d",&m);
while(m--){
getchar();
char c;
scanf("%c",&c);
if(c=='C'){
int idx,val;
scanf("%d%d",&idx,&val);
update(0,0,n-1,idx-1,val);
}
else{
int L,R;
scanf("%d%d",&L,&R);
printf("%d\n",query(0,0,n-1,L-1,R-1));
}
}
}
}