洛谷 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
样例说明:
故输出结果14、16
1 #include<iostream> 2 #include<cstring> 3 #include<cstdio> 4 using namespace std; 5 int n,m,a[500010],tarr[500010]; 6 void add(int pos,int del) 7 { 8 9 while(pos<=n) 10 { 11 tarr[pos]+=del; 12 pos+=(pos^(pos-1))&pos; 13 } 14 15 } 16 int Qry(int pos) 17 { 18 int sum=0; 19 while(pos) 20 { 21 sum+=tarr[pos]; 22 pos-=(pos^(pos-1))&pos; 23 } 24 return sum; 25 } 26 int main() 27 { 28 cin>>n>>m; 29 for(int i=1;i<=n;i++) 30 scanf("%d",&a[i]); 31 for(int i=1;i<=n;i++) 32 add(i,a[i]); 33 int flag,x,y; 34 for(int i=1;i<=m;i++) 35 { 36 scanf("%d%d%d",&flag,&x,&y); 37 if(flag==1) add(x,y); 38 else printf("%d\n",Qry(y)-Qry(x-1)); 39 } 40 41 return 0; 42 }
思路:树状数组 背背版~~~