【cogs240】数列操作【线段树】

【问题描述】

 假设有一列数 {Ai }(1 ≤ i ≤ n) ，支持如下两种操作：

(1)将 A k 的值加 D 。（ k, D 是输入的数）

(2) 输出 A s +A s+1 +…+A t 。（ s, t 都是输入的数， S ≤ T ）

根据操作要求进行正确操作并输出结果。

【输入格式】

     输入文件第一行一个整数 n(0<=n<=100000) ， 第二行为 n 个整数，表示 {A i } 的初始值。

第三行为一个整数 m(0<=m<=150000) ，表示操作数。 下接 m 行，每行描述一个操作，有如下两种情况：

ADD k d ( 表示将 A k 加 d ， 1<=k<=n ， d 为整数 )

SUM s t （表示输出 A s +…+A t ）

【输出格式】

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

【输入输出样例】
 
输入:

4
1 4 2 3 
3
SUM 1 3 
ADD 2 50
SUM 2 3

输出:

7
56

题解:线段树裸题。

#include<iostream>
#include<cstdio>
using namespace std;
int tree[1000001],n,a[1000001],m,s,t,k,d;
char ch[101];
void build(int k,int l,int r)
{
	int mid;
	if (l==r){tree[k]=a[l];return;}
	mid=(l+r)/2;
	build(k*2,l,mid);
	build(k*2+1,mid+1,r);
	tree[k]=tree[k*2]+tree[k*2+1];
}
void add(int k,int l,int r,int a,int d)
{
	int mid;
	if (l==r&&r==a){tree[k]+=d;return;}
    mid=(l+r)/2;
    if (a<=mid) add(k*2,l,mid,a,d);
    else add(k*2+1,mid+1,r,a,d);
    tree[k]=tree[k*2]+tree[k*2+1];
}
int query(int k,int l,int r,int ll,int rr)
{
	int mid,ans(0);
	if (ll<=l&&r<=rr){return tree[k];}
	mid=(l+r)/2;
	if (ll<=mid) ans+=query(k*2,l,mid,ll,rr);
	if (rr>mid)  ans+=query(k*2+1,mid+1,r,ll,rr);
	return ans;
} 
int main()
{
	freopen("shulie.in","r",stdin);
	freopen("shulie.out","w",stdout);
	scanf("%d",&n);
	if (n==0) 
	  {
		return 0;
	  }
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	build(1,1,n);
	scanf("%d",&m);
	for (int i=1;i<=m;i++)
	  {
		cin>>ch; 
		if (ch[0]=='S')
		{
		  scanf("%d%d",&s,&t);
		  printf("%d\n",query(1,1,n,s,t)); 
	    }
	   if (ch[0]=='A')
	     {
			scanf("%d%d",&k,&d);
			add(1,1,n,k,d);
		 }
	  }
}