HDU 4027 Can you answer these queries?

 

http://acm.hdu.edu.cn/showproblem.php?pid=4027

线段树成段更新的题目，开方次数有限，一个区间都是1就lazy。

仿hh风格的线段树，真漂亮啊

#include <iostream>
#include <cmath>
using namespace std ;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=100002 ;
typedef __int64 LL ;
LL sum[maxn<<2] ;
int lazy ;
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1] ;
}
void pushdown(int rt,int m)//区间上的点都到1，不进行更新 
{
    if(sum[rt]==m)
        lazy=1 ;
    else
        lazy=0 ;
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%I64d",&sum[rt]) ;
        return ;
    }
    int m=(l+r)>>1 ;
    build(lson) ;
    build(rson) ;
    pushup(rt) ;
}
void update(int L,int R,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=LL(sqrt((double)sum[rt])) ;
        return ;
    }
    pushdown(rt,r-l+1) ;
    if(lazy)
        return ;
    int m=(l+r)>>1 ;
    if(L<=m)
        update(L,R,lson) ;
    if(R>m)
        update(L,R,rson) ;
    pushup(rt) ;
}
LL query(int L,int R,int l,int r,int rt)
{
    if(L<=l && R>=r){
        return sum[rt] ;
    }
    int m=(l+r)>>1 ;
    LL ans=0 ;
    if(L<=m)    
        ans+=query(L,R,lson) ;
    if(R>m)
        ans+=query(L,R,rson) ;
    return ans ;
}
int main()
{
    int n ;
    int nCase=1 ;
    while(~scanf("%d",&n))
    {
        build(1,n,1) ;
        char op[2] ;
        int m ;
        scanf("%d",&m) ;
        printf("Case #%d:\n",nCase++) ;
        while(m--)
        {
            int a,b ;
            lazy=0 ;
            scanf("%s%d%d",op,&a,&b) ;
            if(a>b)a^=b^=a^=b ; 
            if(op[0]=='0')
                update(a,b,1,n,1) ;
            else
                printf("%I64d\n",query(a,b,1,n,1)) ;
        }
        putchar('\n') ;
    }
    return 0 ;
}