hdu 1166 敌兵布阵（线段树基础题）

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一类的
#define MAX 50005
#define INF 0x7FFFFFFF
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
# define eps 1e-5
//#pragma comment(linker, "/STACK:36777216") ///传说中的外挂
using namespace std;
struct node
{
int left,right,mid,value,cover;
} edge[4*MAX];

void build(int l,int r,int num)
//l，r为当前节点的左右端点，num为当前节点在数组中的下标
{
edge[num].left = l;
edge[num].right = r;
edge[num].value = 0;
edge[num].mid = (l+r) >> 1;
if(l != r) // 如果不是叶子节点
{
build(l,edge[num].mid,num * 2); //左子树
build(edge[num].mid + 1,r,num*2+1); //右子树
}
}
void update(int num,int mid,int v)
//单点更新 mid为更新点，v为更新值
{
if(edge[num].left == mid && edge[num].right == mid)
{
edge[num].value += v;
return ;
}
if(edge[num].mid >= mid)
update(2*num,mid,v);
else
update(2*num+1,mid,v);
edge[num].value = edge[num*2].value + edge[num*2+1].value; // 注意是‘=’不是‘+=’，更新值
}

int query(int l,int r,int num)
{
if(l <= edge[num].left && r >= edge[num].right)
{
return edge[num].value;
}
if(r <= edge[num].mid)
return query(l,r,num*2);
else if(l >= edge[num].mid + 1)
return query(l,r,num *2+1);
else
{
return query(l,edge[num].mid,num*2) + query(edge[num].mid+1,r,num*2+1);
}
}
/*
void insert(int l,int r ,int num)
//l,r为插入段的左右端点，num为当前结点的下标
{
if(l <= edge[num].left && r >= edge[num].right)
//插入段完全大于当前段
{
edge[num].cover = 1;
return ;
}
if(r <= edge[num].mid) //插入段被当前段的左子树包含
insert(l,r,2 * num);
else if(l >= edge[num].mid + 1)//插入段被当前段的右子树包含
insert(l,r,2*num+1);
else
{
insert(l,edge[num].mid, 2* num);
insert(edge[num].mid+1,r,2*num+1);
}
}
*/
int main()
{
int t,i,j;
int casee = 1;
char str[10];
cin >> t;
while(t--)
{
printf("Case %d:\n",casee++);
int n,a;
cin >> n;
build(1,n,1);
for(i=1; i<=n; i++)
{
scanf("%d",&a);
update(1,i,a);
}
while(scanf("%s",str))
{
if(str[0] == 'E')
break;
if(str[0] == 'Q')
{
scanf("%d%d",&i,&j);
printf("%d\n",query(i,j,1));
}
if(str[0] == 'A')
{
scanf("%d%d",&i,&j);
update(1,i,j);
}
if(str[0] == 'S')
{
scanf("%d%d",&i,&j);
update(1,i,-j);
}
}
}
return 0;
}