题目描述
如题,已知一个数列,你需要进行下面两种操作:
-
将某一个数加上 x
-
求出某区间每一个数的和
输入格式
第一行包含两个正整数 n,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 nn 个用空格分隔的整数,其中第 i 个数字表示数列第 i 项的初始值。
接下来 mm 行每行包含 3 个整数,表示一个操作,具体如下:
-
1 x k
含义:将第 x 个数加上 k -
2 x y
含义:输出区间 [x,y] 内每个数的和
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
输入输出样例
输入 #1复制
5 5 1 5 4 2 3 1 1 3 2 2 5 1 3 -1 1 4 2 2 1 4
输出 #1复制
14 16
说明/提示
【数据范围】
对于 30% 的数据,1≤n≤8;
对于 70% 的数据,1≤n,m≤104;
对于 100% 的数据,1≤n,m≤5×105。
数据保证对于任意时刻,a 的任意子区间(包括长度为 1 和 n 的子区间)和均在 [−231,231)范围内。
样例说明:
故输出结果14、16
代码实现:
#include<bits/stdc++.h>
using namespace std;
#define lowbit(x) (x & -(x))
const int N=500010;
int tree[N];
int n,m;
void update(int x,int k)
{
while(x<=n)
{
tree[x]+=k;
x+=lowbit(x);
}
}
int sum(int x)
{
int ans=0;
while(x>0)
{
ans+=tree[x];
x-=lowbit(x);
}
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin>>n>>m;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
update(i,x);
}
while(m--)
{
int op;
cin>>op;
if(op==1)
{
int x,k;
cin>>x>>k;
update(x,k);
}
else
{
int x,y;
cin>>x>>y;
cout<<sum(y)-sum(x-1)<<endl;
}
}
return 0;
}