【luogu 3374,3368】【模板】树状数组1,2
树状数组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
样例说明：
故输出结果14、16
1 #include<cstdio> 2 #include<iostream> 3 #include<cstring> 4 #include<algorithm> 5 #define maxn 500005 6 using namespace std; 7 int n,m,a[maxn],c[maxn]; 8 inline int lowbit(int x){return x&(-x);} 9 inline void add(int pos,int x){ 10 while(pos<=n){ 11 c[pos]+=x; 12 pos+=lowbit(pos); 13 } 14 } 15 inline int getsum(int pos){ 16 int tmp=0; 17 while(pos>0){ 18 tmp+=c[pos]; 19 pos-=lowbit(pos); 20 } 21 return tmp; 22 } 23 int main(){ 24 scanf("%d%d",&n,&m); 25 for(int i=1;i<=n;i++){ 26 scanf("%d",&a[i]); 27 add(i,a[i]); 28 } 29 int a,b,c; 30 for(int i=1;i<=m;i++){ 31 scanf("%d%d%d",&a,&b,&c); 32 if(a==1) add(b,c); 33 else if(a==2) printf("%d\n",getsum(c)-getsum(b-1)); 34 } 35 return 0; 36 }
树状数组2
题目描述
如题，已知一个数列，你需要进行下面两种操作：
1.将某区间每一个数数加上x
2.求出某一个数的和
输入输出格式
输入格式：
第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。
接下来M行每行包含2或4个整数，表示一个操作，具体如下：
操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k
操作2： 格式：2 x 含义：输出第x个数的值
输出格式：
输出包含若干行整数，即为所有操作2的结果。
输入输出样例
输入样例#1：
5 5 1 5 4 2 3 1 2 4 2 2 3 1 1 5 -1 1 3 5 7 2 4
输出样例#1：
6 10
说明
时空限制：1000ms,128M
数据规模：
对于30%的数据：N<=8，M<=10
对于70%的数据：N<=10000，M<=10000
对于100%的数据：N<=500000，M<=500000
样例说明：
故输出结果为6、10
1 #include<cstdio> 2 #include<iostream> 3 #include<cstring> 4 #include<algorithm> 5 #define maxn 500005 6 using namespace std; 7 int n,m,a[maxn],c[maxn],d[maxn]; 8 inline int lowbit(int x){return x&(-x);} 9 inline void add(int pos,int x){ 10 while(pos<=n){ 11 c[pos]+=x; 12 pos+=lowbit(pos); 13 } 14 } 15 inline int getsum(int pos){ 16 int tmp=0; 17 while(pos>0){ 18 tmp+=c[pos]; 19 pos-=lowbit(pos); 20 } 21 return tmp; 22 } 23 int main(){ 24 scanf("%d%d",&n,&m); 25 for(int i=1;i<=n;i++){ 26 scanf("%d",&a[i]); 27 d[i]=a[i]-a[i-1]; 28 add(i,d[i]); 29 } 30 int q,x,y,k; 31 for(int i=1;i<=m;i++){ 32 scanf("%d",&q); 33 if(q==1){ 34 scanf("%d%d%d",&x,&y,&k); 35 d[x]+=k;add(x,k); 36 d[y+1]-=k;add(y+1,-k); 37 } 38 else if(q==2){ 39 scanf("%d",&x); 40 printf("%d\n",getsum(x)); 41 } 42 } 43 return 0; 44 }
