hdu 4027 Can you answer these queries?

题意：给出N个数，有两个操作，一个是把每个数变成原来的数开根号，另一个是查询区间值的和。

思路：这个很明显是线段树。开始的时候我写的是成段更新，结果T了……后来仔细一想，每个数最多开了6次根号就会变成1，变成1以后就不会再变化了，这样的话，即使单点更新，每个数最多更新6次，所以不会超时，但如何保证所有值变成1以后不会再更新了呢？我是用一个数组记录当前区间中1的个数，当这个区间中1的个数等于区间的长度时，就说明这个区间中都是1了，就不用再更新了。这就避免了把更新到1的节点再次更新的问题了。另外，如果不仔细看题的话，这题还有个小坑，那就是给出的x、y并不一定是顺序给的，y可能比x小，因此要事先判断一下，因为这个wrong了好几次，唉。

 

代码：

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
ll sum[maxn<<2];
int fone[maxn<<2];
void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    fone[rt]=fone[rt<<1]+fone[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%I64d",&sum[rt]);
        fone[rt]=(sum[rt]==1)?1:0;
        if(sum[rt]==0) fone[rt]=1;
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    PushUp(rt);
}
void Update(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R&&fone[rt]==r-l+1)
    {
        return;
    }
    if(l==r)
    {
        sum[rt]=(ll)sqrt((double)sum[rt]);
        if(sum[rt]==1) fone[rt]=1;
        return;
    }
    int m=(l+r)>>1;
    if(m>=L&&fone[rt<<1]!=m-l+1)
     Update(L,R,l,m,rt<<1);
    if(m<R&&fone[rt<<1|1]!=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)
{
    if(l>=L&&r<=R)
    {
        return sum[rt];
    }
    int m=(l+r)>>1;
    ll s=0;
    if(m>=L)
      s+=Query(L,R,l,m,rt<<1);
    if(m<R)
      s+=Query(L,R,m+1,r,rt<<1|1);
    return s;
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int n,m;
    int tcase=0;
    while(~scanf("%d",&n))
    {
        tcase++;
        build(1,n,1);
        printf("Case #%d:\n",tcase);
        scanf("%d",&m);
        while(m--)
        {
            int T,x,y;
            scanf("%d%d%d",&T,&x,&y);
            if(x>y) swap(x,y);
            if(T==0)
              Update(x,y,1,n,1);
            else
            {
                ll ans=Query(x,y,1,n,1);
                printf("%I64d\n",ans);
            }
        }
        printf("\n");
    }
    return 0;
}