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