hdu 1166

题目地址:敌兵布阵

按照胡浩版的线段树代码写的

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>
using namespace std;

#define lson l, mid, rt<<1 //直接定义子节点,因为每次都要用到,所以直接定义一个很方便
#define rson mid+1, r, rt<<1|1
const int MAXN = 51000;
int sum[MAXN<<2];
void PushUP(int rt)//向上更新父节点的值
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l, int r, int rt)//建立二叉树,rt即为sum数组下标
{
    if(l == r){//已经到达最底端的叶子节点,即单点,直接输入该值
        scanf("%d", &sum[rt]);
        return;
    }
    int mid = (l+r)>>1;
    build(lson);//向左子节点继续建立二叉树
    build(rson);//向右子节点继续建立二叉树
    PushUP(rt);//全部建立完后向上更新父节点的值
}
void update(int x,int p, int l, int r, int rt)//单点修改
{
    if(l == r){//说明已经到了最低端的叶子节点,已经是单点了,可以直接修改该值
        sum[rt] += p;
        return;
    }
    int mid = (l+r)>>1;
    if(mid>=x)//如果要修改的值在这个区间左边,就进入左子节点继续寻找
        update(x, p, lson);
    else if(mid<x)//如果要修改的值在这个区间右边,就进去右子节点继续寻找
        update(x, p, rson);
    PushUP(rt);//修改完后,仍然要向上更新父节点的值
}
int query(int ll, int rr, int l, int r, int rt)//区间查询
{
    if(ll<=l&&r<=rr){//如果要查询的区间完全覆盖了该子节点,直接返回该子节点的值
        return sum[rt];
    }
    int mid = (l+r)>>1;
    int ans = 0;
    if(mid>=ll) ans+=query(ll,rr,lson);//如果在该子节点左边还有一部分要查询的区间,就去左子树继续查询
    if(mid<rr) ans+=query(ll,rr,rson);//如果在该子节点右边还有一部分要查询的区间,就去右子树继续查询

    return ans;
}
int main()
{
    int t;
    scanf("%d", &t);
    int num = 0;
    int n;
    int a,b;
    char s[10];
    int ans;
    while(t--){
        num++;
        memset(sum,0,sizeof(sum));
        scanf("%d", &n);
        build(1,n,1);//建立
        printf("Case %d:\n", num);
        while(scanf("%s", s)){
            if(s[0] == 'E')break;
            else if(!strcmp(s,"Add")){//修改
                scanf("%d %d", &a, &b);
                update(a,b,1,n,1);
            }else if(!strcmp(s,"Sub")){//修改
                scanf("%d %d", &a, &b);
                update(a,-b,1,n,1);
            }else if(!strcmp(s,"Query")){//查询
                scanf("%d %d", &a, &b);
                ans = query(a,b,1,n,1);
                printf("%d\n", ans);
            }
        }
    }


    return 0;
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值