首页 > 学院 > 开发设计 > 正文

hdu 1166

2019-11-08 01:05:55
字体:
来源:转载
供稿:网友

题目地址:敌兵布阵

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

#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|1const 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;}


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表