【模板】树状数组 1

题目描述

如题,已知一个数列,你需要进行下面两种操作:

1.将某一个数加上x

2.求出某区间每一个数的和

输入输出格式

输入格式:

第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3个整数,表示一个操作,具体如下:

操作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.每个结点c[x]保存以它为根的子树所有叶节点的和

 2.每个节点c[x]的子节点个数等于lowbit(x)的大小

 3.每个内部节点c[x]的父节点是c[x+lobit(x)]

  4.从一个子树根结点到另一个x-lowbit(x)

  其实lowbit(x)可以表示这棵子树的宽度

【code】

 1 #include <cstdio>
 2 #include <iostream>
 3 #include <algorithm>
 4 using namespace std;
 5 int n,m;
 6 int a,c[500005];
 7 inline int lowbit(int x){
 8     return x&(-x);
 9 }
10 inline int GetSum(int x){
11     int ans=0;
12     for(register int i=x;i!=0;i-=lowbit(i))
13         ans+=c[i];
14     return ans;
15 }
16 inline void Add(int x,int y){
17     for(register int i=x;i<=n;i+=lowbit(i))
18         c[i]+=y;
19     return ;
20 }
21 int main(){
22     //freopen("3374.in","r",stdin);
23     //freopen("3374.out","w",stdout);
24     scanf("%d%d",&n,&m); 
25     for(register int i=1;i<=n;i++){
26         scanf("%d",&a);
27         Add(i,a);
28     }
29     int x,y,z;
30     for(register int i=1;i<=m;i++){
31         scanf("%d%d%d",&x,&y,&z);
32         if(x==1)Add(y,z);
33         if(x==2)printf("%d\n",GetSum(z)-GetSum(y-1));
34     }
35     return 0;
36 }

 

posted @ 2019-07-16 16:11  GTR_PaulFrank  阅读(130)  评论(0编辑  收藏  举报