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

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",&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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值