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

来源：互联网发布：域名年费多少钱编辑：程序博客网时间：2024/05/20 14:16

【问题描述】

假设有一列数 {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); }  }}

0 0