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<cstdio> 3 #include<cstring> 4 using namespace std; 5 const int MAXN=500001; 6 int n,m,p; 7 int tree[MAXN];// 8 int lowbit(int p) 9 { 10 return p&(-p); 11 } 12 void point_increase(int w,int v) 13 { 14 for(int i=w;i<=n;i=i+lowbit(i)) 15 tree[i]=tree[i]+v; 16 return ; 17 }//将增加的值加到每一个管理点上 18 int interval_ask(int x) 19 { 20 int ans=0; 21 for(int i=x;i!=0;i=i-lowbit(i)) 22 ans=ans+tree[i]; 23 return ans; 24 }//求出一个前缀和并返回 25 int main() 26 { 27 scanf("%d%d",&n,&m); 28 for(int i=1;i<=n;i++) 29 { 30 scanf("%d",&p); 31 point_increase(i,p); 32 }// 将每一个点初始化为管理自身的管理点 33 for(int i=1;i<=m;i++) 34 { 35 scanf("%d",&p); 36 if(p==1)// 加 37 { 38 int x,y; 39 scanf("%d%d",&x,&y); 40 point_increase(x,y); 41 } 42 else// 求和 43 { 44 int x,y; 45 scanf("%d%d",&x,&y); 46 printf("%d\n",interval_ask(y)-interval_ask(x-1)); 47 }//因为树状数组只能求前缀和所以需要两区间相加 48 } 49 return 0; 50 }