NKOI 1321 数列操作问题

数列操作问题
Time Limit:10000MS  Memory Limit:65536K

Total Submit:345 Accepted:185
Case Time Limit:1000MS
Description
假设有一列数{Ai}(1≤i≤n)，支持如下两种操作：

将Ak的值加D。（k, D是输入的数）

输出As+As+1+…+At。（s, t都是输入的数，S≤T）

Input
第一行一个整数n，

第二行为n个整数，表示{Ai}的初始值≤10000。

第三行为一个整数m，表示操作数

下接m行，每行描述一个操作，有如下两种情况：

ADD k d (表示将Ak加d，1<=k<=n，d为数,d的绝对值不超过10000)

SUM s t （表示输出As+…+At）

Output
对于每一个SUM提问，输出结果

Sample Input

5
1 2 3 2 4
5
SUM 1 2
SUM 1 5
ADD 1 2
SUM 1 2
SUM 1 5

Sample Output

3
12
5
14

Hint
M,N<=100000

线段树的基本题型，具体看代码#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=100005;
struct wk{int a,b,left,right,v;}tree[2*maxn];//a,b分别表示当前节点的左右端点，left,right表示当前节点的左右儿子节点标号，v为权值
int n,m,tot,s,t,k,d,sum[maxn];
char qust[4];
inline void _read(int &x){
char t=getchar();bool sign=true;
while(t<'0'||t>'9')
{if(t=='-')sign=false;t=getchar();}
for(x=0;t>='0'&&t<='9';t=getchar())x=x*10+t-'0';
if(!sign)x=-x;
}
void insert(int x,int y){
int now=++tot;
tree[now].a=x;
tree[now].b=y;
tree[now].v=sum[y]-sum[x-1];//为每一个节点附上权值
if(x<y){
tree[now].left=tot+1;
insert(x,(x+y)/2);
tree[now].right=tot+1;
insert((x+y)/2+1,y);
}
}//建树代码
void add(int r,int d){
tree[r].v+=d;
if(tree[r].left!=0&&tree[tree[r].left].a<=k&&tree[tree[r].left].b>=k)
add(tree[r].left,d);
if(tree[r].right!=0&&tree[tree[r].right].a<=k&&tree[tree[r].right].b>=k)
add(tree[r].right,d);
}//ADD操作函数
int getsum(int r){
if(tree[r].a>=s&&tree[r].b<=t)return tree[r].v;//如果当前节点是要求的区间的子集，则返回当前节点表示区间的和
else{//讨论两段区间相交的情况
int total=0;
//这里的相交有三种情况，要么左端点在区间内，要么右端点在区间内，要么是当前的讨论的区间包含了要求的区间
 if(tree[r].left!=0&&((tree[tree[r].left].a<=t&&tree[tree[r].left].a>=s)||(tree[tree[r].left].b>=s&&tree[tree[r].left].b<=t)||(s>=tree[tree[r].left].a&&t<=tree[tree[r].left].b)))
total+=getsum(tree[r].left);
if(tree[r].right!=0&&((tree[tree[r].right].a<=t&&tree[tree[r].right].a>=s)||(tree[tree[r].right].b>=s&&tree[tree[r].right].b<=t)||(s>=tree[tree[r].right].a&&t<=tree[tree[r].right].b)))
total+=getsum(tree[r].right);
return total;
}
}//SUM操作函数
int main(){
int i,x;
scanf("%d",&n);
for(i=1;i<=n;i++){
_read(x);
sum[i]=sum[i-1]+x;
}
insert(1,n);
scanf("%d",&m);
for(i=1;i<=m;i++){
scanf("%s",qust);
if(qust[0]=='A'){
_read(k);_read(d);
add(1,d);
}
else {
_read(s);_read(t);
printf("%d\n",getsum(1));
}
}
}