Can you answer these queries? HDU - 4027 区间修改&&（优化）

Can you answer these queries? HDU - 4027

题意

给定n个数，分别代表每一个战舰的耐力值。有m个操作，当 t=1时，输出x~y的战舰耐力值之和，当 t=0时，使x~y战舰的每一个耐力值都变为原来的平方根（向下取整）。

思路

遍历 x到y 的每一个叶子节点，使他们的值变为原来的平方根，结果超时了。因为 1 的平方根还是 1，所以在更新时判断这段区间的值是否都为1，如果都为1就不需要更新。

一开始以为会用到标记数组，最后发现根本不许需要那么麻烦。。。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=262144;
ll sum[maxn],node[maxn/2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void bulid(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=node[l];
        return ;
    }
    int m=(l+r)>>1;
    bulid(l,m,rt<<1);
    bulid(m+1,r,rt<<1|1);
    pushup(rt);
}
void update(int L,int R,int l,int r,int rt)  //更新 L~R区间
{
    if(l==r)
    {
        sum[rt]=sqrt(sum[rt]);
        return ;
    }
    if(L<=l && R>=r && sum[rt]==r-l+1)  //优化  判断区间的值是否都为1
        return ;
    int m=(l+r)>>1;
    if(L<=m)
        update(L,R,l,m,rt<<1);
    if(R>m)
        update(L,R,m+1,r,rt<<1|1);
    pushup(rt);
}
ll query(int L,int R,int l,int r,int rt)  // L~R区间求和
{
    if(L<=l && R>=r)
        return sum[rt];
    ll cnt=0;
    int m=(l+r)>>1;
    if(L<=m)
        cnt+=query(L,R,l,m,rt<<1);
    if(R>m)
        cnt+=query(L,R,m+1,r,rt<<1|1);
    return cnt;
}
int main()
{
    int n,m,k=1;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            scanf("%lld",&node[i]);
        bulid(1,n,1);
        scanf("%d",&m);
        int x,y,z;
        printf("Case #%d:\n",k++);
        while(m--)
        {
            scanf("%d%d%d",&z,&x,&y);
            if(x>y)      //当 x>y时，交换两点的值
                swap(x,y);
            if(z==1)
                printf("%lld\n",query(x,y,1,n,1));
            else
                update(x,y,1,n,1);
        }
        printf("\n");
    }
    return 0;
}