【问题描述】
假设有一列数 {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);
}
}
}