线段树的建树、更新、查询操作

模板题SDUT OJ 3771 数组计算机

Description
bLue 有一个神器的机器，这个机器可以读入一个数组，并按照用户要求快速地进行数组的处理和计算，它支持如下两种操作：

操作 1：把数组中第 p 个元素的值增加 v。
操作 2：计算数组中 [l, r] 区间内所有数的和。
这个机器就是这么的神奇，但是 bLue 的计算机坏掉了，你能帮他修一下吗？

Input
输入数据有多组（数据组数不超过 20），到 EOF 结束。

对于每组数据：

第 1 行输入一个整数 n (1 <= n <= 10^5)，表示数组中元素的个数。
第 2 行输入 n 个用空格隔开的整数 ai (1 <= ai <= 10^10)，表示初始输入到计算机中的数组。
第 3 行输入一个整数 q (1 <= q <= 50000)，表示用户的操作次数。
接下来 q 行，每行输入先输入 1 个整数，表示操作类型，根据不同的操作类型：
如果类型为 1，则紧接着输入 2 个用空格隔开的整数 p (1 <= p <= n) 和 v (1 <= v <= 10^10)，表示要把数组中第 p 个数的值增加 v。
如果类型为 2，则紧接着输入 2 个用空格隔开的整数 l, r (1 <= l <= r <= n)，表示要计算区间 [l, r] 内所有数的和（数组下标从 1 开始）。
Output
对于每组数据中的每次类型为 2 的操作，输出 1 行，包含一个整数，表示计算出的和。
Sample
Input

5
1 2 3 4 5
5
2 1 2
2 1 5
1 4 10
2 4 5
2 1 5

Output

3
15
19
25

Accepted代码：

#include<bits/stdc++.h>

using namespace std;

long long arr[100000];
long long  tree[400000];

void build_tree(int node, int start, int end) {		//建树操作
	if (start == end) {								//递归出口
		tree[node] = arr[start];
		return;						//记得return； 不然此函数递归的时候跳不出来
	}

	int mid = (start + end) / 2;
	int left_node = 2 * node + 1;
	int right_node = 2 * node + 2;

	build_tree(left_node, start, mid);		//递归建树  左分支和右分支
	build_tree(right_node, mid + 1, end);

	tree[node] = tree[left_node] + tree[right_node];//递归完后两边分支的值加到一个数的节点上

	return;									//不知道有没有用，但是一般给调用的函数加一个return；让其停止调用，防止调用的时候跳不出来
}

void update(int node, int start, int end, int index, long long  value) {		//更新树的每个节点
	if (start == end) {													//index是要更新的数组下标  value是要更新掉的数组的值 都还跟树里的东西没关
		arr[index] += value;										//递归调用出口
		tree[node] = arr[index];									//这时候要把树的节点值给更新掉了
		return;														//调用结束记得return;
	}

	int mid = (start + end) / 2;
	int left_node = 2 * node + 1;
	int right_node = 2 * node + 2;

	if (index >= 0 && index <= mid) {								//寻找左分支还是右分支
		update(left_node, start, mid, index, value);				//递归更新分支的值
	}
	else {
		update(right_node, mid + 1, end, index, value);
	}

	tree[node] = tree[left_node] + tree[right_node];				//由树的下边往上更新每一个节点的值

	return;
}

long long search(int node, int start, int end, int l, int r) {	//线段树的查询操作
	if (r<start || l>end) {										//递归调用出口
		return 0;
	}
	else if (l <= start && end <= r) {							//这里注意一下，这里的条件很重要
		return tree[node];										//这里是为了让找到这个区间后直接就返回树的这个节点值
	}														 	 //如果不用这个条件判断，后边会一直计算到每一个叶子节点（做无用功）
	else if (start == end) {
		return tree[node];
	}
	else {

		int mid = (start + end) / 2;
		int left_node = 2 * node + 1;
		int right_node = 2 * node + 2;
		long long  sum_left = search(left_node, start, mid, l, r);			//long long 跟这个题有关，我因为数据类型的问题错了很多次
		long long  sum_right = search(right_node, mid + 1, end, l, r);

		return sum_left + sum_right;
	}
}

int main() {
	std::ios::sync_with_stdio(false);
	int n;
	while (cin >> n) {
		for (int i = 0; i < n; i++) {
			cin >> arr[i];
		}
		build_tree(0, 0, n - 1);
		int t;
		cin >> t;
		while (t--) {
			int operation;
			int p, l, r;
			long long v;					//long long 因为这个数据类型的原因我错了很多次
			cin >> operation;
			if (operation == 1) {
				cin >> p >> v;
				update(0, 0, n - 1, p - 1, v);
			}
			else {
				cin >> l >> r;
				printf("%lld\n", search(0, 0, n - 1, l - 1, r - 1));
			}
		}
	}
	return 0;
}

学习线段树还得看 灯神的视频
强烈推荐，这个视频太强了，看了一遍大彻大悟