南阳oj116--士兵杀敌（二）（线段树，更新，求和）

士兵杀敌（二）

点击打开链接

描述
南将军手下有N个士兵，分别编号1到N，这些士兵的杀敌数都是已知的。
小工是南将军手下的军师，南将军经常想知道第m号到第n号士兵的总杀敌数，请你帮助小工来回答南将军吧。
南将军的某次询问之后士兵i可能又杀敌q人，之后南将军再询问的时候，需要考虑到新增的杀敌数。

输入
只有一组测试数据
第一行是两个整数N,M，其中N表示士兵的个数(1<N<1000000)，M表示指令的条数。(1<M<100000)
随后的一行是N个整数，ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行是一条指令，这条指令包含了一个字符串和两个整数，首先是一个字符串，如果是字符串QUERY则表示南将军进行了查询操作，后面的两个整数m,n，表示查询的起始与终止士兵编号；如果是字符串ADD则后面跟的两个整数I,A(1<=I<=N,1<=A<=100),表示第I个士兵新增杀敌数为A.

输出
对于每次查询，输出一个整数R表示第m号士兵到第n号士兵的总杀敌数，每组输出占一行

样例输入
5 6
1 2 3 4 5
QUERY 1 3
ADD 1 2
QUERY 1 3
ADD 2 3
QUERY 1 2
QUERY 1 5

样例输出
6
8
8
20
来源
[张云聪]原创
上传者
张云聪

解题思路：比杀敌一多了一个单点更新，输入字符串判断时，只要判断第一个字符就可以啦。

#include<cstdio>
#include<algorithm>
using namespace std;
#define L o<<1         //乘二 
#define R (o<<1)|1     //乘二加一 
#define MAX 1000000
struct Node
{
	int l,r,sum;
}Tree[MAX<<2];    //将数组乘四，记住 
void PushUp(int o)
{
	Tree[o].sum = Tree[o*2].sum + Tree[o*2+1].sum;
//	Tree[o].Max = max(Tree[o*2].Max,Tree[o*2+1].Max);
//	Tree[o].Min = min(Tree[o*2].Min,Tree[o*2+1].Min);
}
void Build(int o,int l,int r)    // o代表节点位置，l为左边界，r为右边界 
{
	//首先记录l和r的值 
	Tree[o].l = l;
	Tree[o].r = r;
	if (l == r)		//到达最底层，递归终止
	{
		int t;
		scanf ("%d",&t);		//输入数据 
		Tree[o].sum /*= Tree[o].Max = Tree[o].Min */= t;		//更新节点数据 
		return;
	}
	int mid = (l+r) >> 1;		//找到中间节点 
	Build(o*2 , l , mid);		//递归建左子树 
	Build(o*2+1 , mid+1 , r);		//递归建右子树 
	PushUp(o);		//更新当前节点的值 
}
void UpDate(int o,int x,int y)	  //更新单点
{
	if (Tree[o].l == Tree[o].r )		//递归结束
	{
		Tree[o].sum += y;		//精确找到了节点，更新 
		return;
	}
	int mid = (Tree[o].l+Tree[o].r) / 2;		//找到中间位置
	if (x <= mid)
		UpDate(o*2,x,y);		//找左子树 
	else
		UpDate(o*2+1,x,y);		//找右子树 
	PushUp(o);		//更新当前节点 
}
int Query(int o,int l,int r)        //查询  
{  
    if (Tree[o].l == l && Tree[o].r == r)  
        return Tree[o].sum;  
    int mid = (Tree[o].l + Tree[o].r) >> 1;  
    if (mid >= r)  
        return Query(o << 1 , l , r);     //去左孩子找   
    else if (l > mid)  
        return Query(o << 1 | 1 , l ,r);      //去右孩子找  
    else  
        return (Query(o << 1 , l , mid) + Query(o << 1 | 1 , mid + 1 ,r));      //两边寻找   
}  
int main()
{
	int n,m,x,y;
	scanf ("%d %d",&n,&m);
	Build(1,1,n);      //1,1是固定的 
	while(m--)
	{
		char q[8];
		scanf("%s %d %d",q,&x,&y);
		if(q[0]=='Q')
		{
			printf("%d\n",Query(1,x,y));
		}
		else
		UpDate(1,x,y);
	 } 	
	return 0;
}