树状数组(1)
题目描述
如题,已知一个数列,你需要进行下面两种操作:
-
将某一个数加上 x
-
求出某区间每一个数的和
输入格式
第一行包含两个正整数 n,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 n个用空格分隔的整数,其中第 i个数字表示数列第 ii项的初始值。
接下来 m 行每行包含 3 个整数,表示一个操作,具体如下:
-
1 x k
含义:将第 xx个数加上 k -
2 x y
含义:输出区间 [x,y] 内每个数的和
输出格式
输出包含若干行整数,即为所有操作 2的结果。
in:
5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4
out:
14
16
#include<bits/stdc++.h> using namespace std; int tree[500050]; int n,m; int lowbit(int x){ return x&-x; } void add(int x,int k) { for(int i=x;i<=n;i+=lowbit(i)) { tree[i]+=k; } } int find1(int x) { int sum=0; for(int i=x;i;i-=lowbit(i)) { sum+=tree[i]; } return sum; } int main(){ cin>>n>>m; for(int i=1;i<=n;i++) { int x; cin>>x; add(i,x); } for(int i=1;i<=m;i++) { int t,x,y; cin>>t>>x>>y; if(t==1) add(x,y); else cout<<find1(y)-find1(x-1)<<endl; } return 0; }