【解题报告】 NYOJ 116 士兵杀敌（二） 线段树 单点更新求区间和

题目连接： NYOJ 116

这个是线段树的基本操作----更新和查询，和上一个题一样，只不过一个是求最大值一个是求区间和，看来线段树在更新和查询的时候对时间的消耗很小很小。这个属于基础水题，我就不说什么了。

如果不知道线段树的基本操作，请移步 线段树的基本操作

// NYOJ 116 士兵杀敌（二） 线段树 --单点更新求区间和
// 
 
#include <cstdio>
#include <cmath>
#include <iostream>
using namespace std;

const int MAXNODE = 2097152; // == (1 << 21) - 1  depth:21
const int MAXS = 1000003;
struct SOILD{
	int left,right; // 数组模拟满二叉树, 区间 [left,right]
	int value;
}Soldier[MAXNODE];

int father[MAXS]; // 存每一个士兵的父结点

void BuildTree(int i,int left,int right,int devalue){ // 初始值为0
	Soldier[i].left = left;
	Soldier[i].right = right;
	Soldier[i].value = devalue;
	if (left == right){
		father[left] = i; // 可以瞬间知道第i个士兵的父节点是哪个
		return;
	}
	BuildTree(i*2, left, (int)floor((right+left)/2.0), 0);
	BuildTree(i*2+1, (int)floor((right+left)/2.0)+1, right, 0);
}

void UpdataTree(int ri,int iKilln){ // 单点更新: 点p[i,i] 更新为iKilln (初始为0)
	if (ri == 1){
		Soldier[ri].value += iKilln;
		return;
	}
	Soldier[ri].value += iKilln; // iKilln 设为全局变量应该快一点
	UpdataTree(ri/2,iKilln);
}

int res;
void Query(int i,int a,int b){ // [a,b]区间的和
	if (a == Soldier[i].left && b == Soldier[i].right){
		res += Soldier[i].value;
		return ;
	}
	if (a <= Soldier[i*2].right){ // 左端点小于右端点（几何含义：区间横跨过中点） 则区间需要分割
		if (b <= Soldier[i*2].right){
			Query(i*2, a, b); // 左边全包含
		}else{
			Query(i*2, a, Soldier[i*2].right); // 横跨取左
		}
	}
	if (b >= Soldier[i*2+1].left){
		if (a >= Soldier[i*2+1].left){
			Query(i*2+1, a, b); // 右边全包含
		}else{
			Query(i*2+1, Soldier[i*2+1].left, b); // 横跨取右
		}
	}
}

int main(){
	int n_s,n_q,iKilln,a,b;
	cin >> n_s >> n_q;
	BuildTree(1,1,n_s,0);
	for (int i = 1 ;i <= n_s; i++){ // 建树
		scanf("%d",&iKilln);	// iKilln 为第 i 个增加的杀敌数
		UpdataTree(father[i],iKilln); // 瞬间找到士兵的结点序号
	}
	char order[10];
	for (int i = 1; i <= n_q; i++){
		scanf("%s %d %d",&order[0],&a,&b); // [a,b] ; a + kill b 	
		if (order[0] == 'Q'){
			res = 0;
			Query(1,a,b);
			printf("%d\n", res);
		}else
			UpdataTree(father[a],b);
	}
	return 0;
}