线段树构建、更新及计算区间和

已知一个数组，要求能够快速的计算给定left和right之间的和可以用线段树。

例如：已知arr = [1, 3, 5, 7, 9, 11]，求索引在[2, 5]中的所有数的和。该例子中，arr为原数组， left=2, right=5。

Ps：以下代码的编译环境是Vs2017，所以需要加system("pause");显示输出，如果是Vs2019，则无需加这行代码。此外，因为Vs较为严谨，程序可能因为一些老的.c文件使用了strcpy,scanf等不安全的函数，而报警告和错误，而导致无法编译通过。
所以加上#define _CRT_SECURE_NO_WARNINGS

//数据结构与算法之线段树
#define _CRT_SECURE_NO_WARNINGS
#define MAX_LEN 100
#include <iostream>
using namespace std;

//构建线段树
/*参数含义
	arr:源数组
	tree:线段树
	node:当前所在线段树结点编号--->可以理解为每次递归的树的根节点所在tree[]数组中的索引
	start:区间左边界
	end:区间右边界
*/
void build_segment(int arr[], int tree[], int node, int start, int end) {
	//递归头--->递归出口
	if (start == end) {//如果区间左边界与右边界重叠，说明此节点是叶子节点，将该节点的值赋值成arr数组在start(或end)索引处的值
		tree[node] = arr[start];
	}
	//递归体
	else {
		//计算中间点，将源区间分成两部分
		int mid = start + (end - start) / 2;
		//计算左右孩子节点编号
		int left_node  = 2 * node + 1;
		int right_node = 2 * node + 2;
		//构建左子线段树和右子线段树
		build_segment(arr, tree, left_node,  start, mid);
		build_segment(arr, tree, right_node, mid+1, end);
		//计算当前节点的值为左右子树根节点值之和
		tree[node] = tree[left_node] + tree[right_node];
	}
}

//更新数组某个值
/*参数列表
	arr:源数组
	tree:线段树
	start:搜索的区间左边界
	end:搜索的区间右边界
	index:要更新的索引
	val:更新成值val
*/
void update(int arr[], int tree[], int node, int start, int end, int index, int val) {
	//递归头
	if (start == end) {//区间左右边界相等
		//更新
		arr[index] = val;
		tree[node] = val;
	}
	//递归体
	else {
		//中间位置
		int mid = start + (end - start) / 2;
		//左孩子
		int left_node = 2 * node + 1;
		//右孩子
		int right_node = 2 * node + 2;
		//index在左区间
		if (index >= start && index <= mid) {
			update(arr, tree, left_node, start, mid, index, val);
		}
		//index在右区间
		else if (index > mid && index <= end) {
			update(arr, tree, right_node, mid + 1, end, index, val);
		}
		//更新tree
		tree[node] = tree[left_node] + tree[right_node];
	}
}

//计算给定区间内元素的和
/*参数：
	arr:源数组
	tree:线段树
	node:当前节点
	start:当前所查结点所代表区间的左边界
	end:当前所查结点所代表的区间的右边界
	L:给定区间左边界
	R:给定区间右边界
*/
int sum(int tree[], int node, int start, int end, int L, int R) {
	//递归头
	if (start > R || end < L) {//当前所查区间与给定区间无交集，直接返回0
		return 0;
	}
	else if (start >= L && end <= R) {//当前所查区间是给定区间子区间，返回当前树节点的值tree[node]
		return tree[node];
	}
	//递归体
	else {
		//中间位置
		int mid = start + (end - start) / 2;
		//左孩子
		int left_node  = 2 * node + 1;
		//右孩子
		int right_node = 2 * node + 2;
		//向左查询+向右查询
		return sum(tree, left_node, start, mid, L, R) + sum(tree, right_node, mid + 1, end, L, R);
	}
}

int main() {
	//源数组
	int arr[] = { 1,3,5,7,9,11 };
	//源数组大小
	int size = 6;
	//线段树(数组形式)
	int tree[MAX_LEN] = { 0 };
	
	//输出源数组
	for (int i = 0; i < size; i++) {
		cout << arr[i] << endl;
	}
	cout << "===============================" << endl;
	//构建线段树
	build_segment(arr, tree, 0, 0, size-1);
	for (int i = 0; i < 15; i++) {
		cout << tree[i] << endl;
	}
	cout << "===============================" << endl;
	//更新线段树
	update(arr, tree, 0, 0, 5, 4, 6);
	for (int i = 0; i < 15; i++) {
		cout << tree[i] << endl;
	}
	//计算给定区间的元素和
	cout << "================================" << endl;
	int rangeSum = sum(tree, 0, 0, 5, 2, 5);
	cout << rangeSum << endl;
	system("pause");
	return 0;
}