首页 > 其他 > 详细

hdu-1166-敌兵布阵-线段树-单点更新,区域查询

时间:2014-05-26 05:25:40      阅读:381      评论:0      收藏:0      [点我收藏+]

线段树的单点更新,区域查询操作。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
using namespace std;
#define lmin 1
#define rmax n
#define lson l,(l+r)/2,rt<<1
#define rson (l+r)/2+1,r,rt<<1|1
#define root lmin,rmax,1
#define maxn 55000
int cl[maxn*4];
int num[maxn];
void push_up(int rt)
{
    cl[rt]=cl[rt<<1]+cl[rt<<1|1];
}
void push_down(int rt)
{
    if(cl[rt]!=-1)
    {
        cl[rt<<1]=cl[rt];
        cl[rt<<1|1]=cl[rt];
        cl[rt]=-1;
    }
}
void creat(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%d",&cl[rt]);
        return;
    }
    creat(lson);
    creat(rson);
    push_up(rt);
}
void update(int ll,int rr,int x,int l,int r,int rt)
{
    if(l>rr||r<ll)return;
    if(ll<=l&&rr>=r)
    {
        cl[rt]+=x;
        return;
    }
   // push_down(rt);
    update(ll,rr,x,lson);
    update(ll,rr,x,rson);
    push_up(rt);
}
int query(int ll,int rr,int l,int r,int rt)
{
    if(r<ll)return 0;
    if(l>rr)return 0;
    if(ll<=l&&rr>=r)return cl[rt];
    return query(ll,rr,lson)+query(ll,rr,rson);
}
int main()
{
    int T,_,n,l,r,st,x;
    char str[110];
    scanf("%d",&T);
    for(_=1;_<=T;_++)
    {
        printf("Case %d:\n",_);
        scanf("%d",&n);
        creat(root);
        while(1)
        {
            scanf("%s",str);
            if(str[0]=='E')break;
            else if(str[0]=='Q')
            {
                scanf("%d%d",&l,&r);
                printf("%d\n",query(l,r,root));
            }
            else
            {
                scanf("%d%d",&st,&x);
                if(str[0]=='A')update(st,st,x,root);
                else update(st,st,-x,root);
            }
        }
    }
    return 0;
}


hdu-1166-敌兵布阵-线段树-单点更新,区域查询,布布扣,bubuko.com

hdu-1166-敌兵布阵-线段树-单点更新,区域查询

原文:http://blog.csdn.net/rowanhaoa/article/details/26743809

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!