您的位置:首页 > 其它

HDU 1166 敌兵布阵(线段树入门,单点更新求和)

2018-02-06 21:20 363 查看
A - 敌兵布阵 

        HDU - 1166
这道题是搜到的线段树模板题,自己模仿大牛的写法,写出来也可以理解,目前自己的水平尚可以解读,感觉比较容易理解,不过以后要靠自己写了,虽然心里比较没谱。。。多多钻研,细节比较多#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
using namespace std;
#define maxn 50005
int sum[maxn << 2];//线段树需要的元素个数一般为四倍左右,需要开四倍空间
int n;
//rt << 1 表示当前节点的左节点, rt << 1 | 1 表示右节点(比左节点大 1)
void pushup(int rt)//更新节点信息,本题即求和
{
sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];//当前节点的值等于其左节点加右节点的值
}
void build(int l, int r, int rt)//递归方法建树
{
if(l == r)
{
scanf("%d",&sum[rt]);
return ;
}
int m = (l + r) >> 1;
build(l, m, rt << 1);
build(m + 1, r, rt << 1 | 1);
pushup(rt);//更新节点值
}
/*
这里 递归的终点就是每个线段都只包含一个点, 即 l == r, 此时共有n个元素需要输入
*/
void update(int p, int add, int l, int r,int rt)//更新数据,需要更新的是包含p元素的节点,需要更新的值为add
{
if(l == r)//递归到线段树最后一层,此时每个区间只包含一个点
{
sum[rt] += add;
return ;
}
int m = (l + r) >> 1;
if(p <= m)update(p, add, l, m, rt << 1);//需要更新的地方
else update(p, add, m + 1, r, rt << 1 | 1);
pushup(rt);//更新
}
int query(int left, int right, int l, int r, int rt)//区间查询函数
{
if(left <= l && right >= r)return sum[rt];//包含于目前的区间之中就加上该节点值
int m = (l + r) >> 1;
int ans = 0;
if(left <= m)ans += query(left, right, l, m, rt << 1);
if(right > m)ans += query(left, right, m + 1, r, rt << 1 | 1);
return ans;
}
int main()
{
int t;
int cas = 1;
char a[10];
scanf("%d",&
4000
;t);
while(t --)
{
scanf("%d",&n);
build(1, n, 1);
printf("Case %d:\n", cas ++);
while(scanf("%s",a)!=EOF)
{
if(a[0] == 'E')break;
int x, y;
scanf("%d%d",&x,&y);
if(a[0] == 'Q')
{
int ans = query(x, y ,1, n, 1);
printf("%d\n",ans);
}
if(a[0] == 'S')update(x, -y, 1, n, 1);
if(a[0] == 'A')update(x, y, 1, n, 1);
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: