【模板】树状数组 1

题目链接

题目描述

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

• 将某一个数加上 x

• 求出某区间每一个数的和

输入格式

第一行包含两个正整数 n,m，分别表示该数列数字的个数和操作的总个数。

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

接下来 m 行每行包含 3 个整数，表示一个操作，具体如下：

• 1 x k 含义：将第 x 个数加上 k

• 2 x y 含义：输出区间 [x,y] 内每个数的和

输出格式

输出包含若干行整数，即为所有操作 2 的结果。

输入输出样例

输入

5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4

输出

14
16

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 10;
int n, m, f[maxn];

int lowbit(int k)
{
	return k & -k;
}

void update(int x, int k)	// 单点更新 
{
	while(x <= n){
		f[x] += k;
		x += lowbit(x);
	}
}

int query(int x)	// 区间查询 
{
	int ans = 0;
	while(x != 0){
		ans += f[x];
		x -= lowbit(x);
	}
	return ans;
}

int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i++){
		int k;
		cin >> k;
		update(i, k);
	}
	while(m--){
		int a, b, c;
		cin >> a >> b >> c;
		if(a == 1){
			update(b, c);
		}
		else if(a == 2){
			int sum = query(c) - query(b-1);
			cout << sum << endl;
		}
	}
	return 0;
}