洛谷 P3374 【模板】树状数组 1 单点修改 区间查询
题目描述
如题,已知一个数列,你需要进行下面两种操作:
将某一个数加上 x
求出某区间每一个数的和
输入格式
第一行包含两个正整数 n,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 n 个用空格分隔的整数,其中第 i 个数字表示数列第 i 项的初始值。
接下来 m 行每行包含 3 个整数,表示一个操作,具体如下:
1 x k 含义:将第 x 个数加上 k
2 x y 含义:输出区间 [x,y] 内每个数的和
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
输入输出样例
输入
5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4
输出
14
16
AC代码:
#include <iostream>
#include <cstdio>
using namespace std;
int tree[500005];
int n,m,a[500005];
void update(int i,int val)
{
while(i<=n)
{
tree[i]+=val;
i+=-i&i;
}
}//树状数组更新
long long getsum(int i)
{
long long sum=0;
while(i>0)
{
sum+=tree[i];
i-=-i&i;
}
return sum;
}//树状数组求和
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
update(i,a[i]);
}
for(int i=1;i<=m;i++)
{
int opt,x,y;
scanf("%d%d%d",&opt,&x,&y);
if(opt==1)
update(x,y);
if(opt==2)
printf("%d\n",getsum(y)-getsum(x-1));
}
return 0;
}