HDU4027 Can you answer these queries? [线段树]

  简单线段树,开根号开几次这个数就为1了,之后再对其开根号就没有意义了。当一个区间全部为0或者1时就没有必要再去跟新这个区间了。

  

#include <string.h>
#include <stdio.h>
#include <math.h>
#define lson l,m,p<<1
#define rson m+1,r,p<<1|1
#define calm (l+r)>>1
#define MAXN 100005

typedef __int64 LL;
int n,q,cas=1,tu,tv,tq;
LL sum[MAXN<<2],x[MAXN];
int root[MAXN<<2];
void pushup(int p){
    root[p]=root[p<<1]+root[p<<1|1];
    sum[p]=sum[p<<1]+sum[p<<1|1];
}
void init(int l,int r,int p){
    if(l==r){
        sum[p]=x[l];
        root[p]=(x[l]>1?1:0);
        return;
    }
    int m=calm;
    init(lson);init(rson);
    pushup(p);
}
void update(int L,int R,int l,int r,int p){
    if(root[p]==0)return;
    if(l==r){
        sum[p]=(LL)sqrt((double)(sum[p]+0.5));
        root[p]=(sum[p]>1?1:0);
        return;
    }
    int m=calm;
    if(L<=m)update(L,R,lson);
    if(R>m)update(L,R,rson);
    pushup(p);
}
LL query(int L,int R,int l,int r,int p){
    if(L<=l&&r<=R)return sum[p];
    LL tmp=0;
    int m=calm;
    if(L<=m)tmp+=query(L,R,lson);
    if(R>m)tmp+=query(L,R,rson);
    return tmp;
}
int main(){
    freopen("test.in","r",stdin);
    while(scanf("%d",&n)!=EOF){
        for(int i=1;i<=n;i++)scanf("%I64d",&x[i]);
        init(1,n,1);
        scanf("%d",&q);
        printf("Case #%d:\n",cas++);
        while(q--){
            scanf("%d%d%d",&tq,&tu,&tv);
            if(tu>tv)tu^=tv,tv=tu^tv,tu=tu^tv;
            if(tq==0){
                update(tu,tv,1,n,1);
            }else{
                printf("%I64d\n",query(tu,tv,1,n,1));
            }
        }
        printf("\n");
    }
}
posted @ 2012-08-27 23:35  Burn_E  阅读(136)  评论(0编辑  收藏  举报