洛谷 P3374 【模板】树状数组 1
题目
题目描述
如题,已知一个数列,你需要进行下面两种操作:
1.将某一个数加上x
2.求出某区间每一个数的和
输入输出格式
输入格式:
第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。
接下来M行每行包含3或4个整数,表示一个操作,具体如下:
操作1: 格式:1 x k 含义:将第x个数加上k
操作2: 格式: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
说明
时空限制:1000ms,128M
数据规模:
对于30%的数据:N<=8,M<=10
对于70%的数据:N<=10000,M<=10000
对于100%的数据:N<=500000,M<=500000
题解
树状数组(最不想打的东西,感觉自己理解有点难QAQ)
关键操作lowbit(
x
&-
(-
x
在二进制下的表达就是对
(不知道的话最好自己打一个代码,找找规律,应该也可以看懂这个操作)
然后就是根据这个性质做一个类似线段树的东西(具体的我也说不来,反正加、乘等操作就是不断将子节点的信息更新到父节点,然后询问操作就是从父节点分解到各个子节点)
代码
#include<cstdio>
using namespace std;
int n,m;
int a[500005],sum[500005];
int readln()
{
int x=0,f=1;
char ch=getchar();
while (ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=getchar();}
while ('0'<=ch&&ch<='9') x=x*10+ch-48,ch=getchar();
return x*f;
}
int lowbit(int x){return x&-x;}
void add(int x,int y)
{
while (x<=n)
{
sum[x]+=y;x+=lowbit(x);
}
}
int get(int x)
{
int ret=0;
while (x>0)
{
ret+=sum[x];
x-=lowbit(x);
}
return ret;
}
int main()
{
n=readln();m=readln();
for (int i=1;i<=n;i++) add(i,readln());
for (int i=1;i<=m;i++)
{
int x=readln(),y;
if (x==1) {
x=readln();y=readln();
add(x,y);
}
else {
x=readln();y=readln();
printf("%d\n",get(y)-get(x-1));
}
}
return 0;
}