BZOJ 3038 上帝造题的七分钟2

Description
XLk觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。
“第一分钟，X说，要有数列，于是便给定了一个正整数数列。
第二分钟，L说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。
第三分钟，k说，要能查询，于是便有了求一段数的和的操作。
第四分钟，彩虹喵说，要是noip难度，于是便有了数据范围。
第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。
第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。
第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。”
——《上帝造题的七分钟·第二部》
所以这个神圣的任务就交给你了。

---

【题目分析】
并查集+树状数组水过去了。
其实线段树也可以。

---

【代码】

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define ll long long
using namespace std;
long long a[100001],f[100001],n,q,op,l,r;
long long t[100001];
inline long long gf(long long k)
{
    if (f[k]==k) return k;
    else return f[k]=gf(f[k]);
}
inline void add(long long x,long long f)
{
    for (;x<=n;x+=x&(-x))
        t[x]+=(ll)f;
}
inline long long gs(long long x)
{
    long long ret=0;
    for (;x;x-=x&(-x)) ret+=t[x];
    return ret;
}
int main()
{
    freopen("ship.in","r",stdin);
    freopen("ship.out","w",stdout);
    scanf("%lld",&n);
    for (long long i=1;i<=n;++i) scanf("%lld",&a[i]),f[i]=i;
    f[n+1]=n+1;
    for (long long i=1;i<=n;++i) add(i,a[i]);
    scanf("%lld",&q);
    for (long long zz=1;zz<=q;++zz)
    {
        scanf("%lld",&op);
        if (op==0)
        {
            scanf("%lld%lld",&l,&r);
            if (l>r) swap(l,r);
            long long i=l;
            while (i<=r)
            {
//              printf(" mod on %lld\n",i);
                i=gf(f[i]);
                if (i>r) break;
//              printf(" mod on %lld\n",i);
                long long tmp=a[i];
                a[i]=(long long)sqrt(a[i]);
                add(i,a[i]-tmp);
                if (a[i]==1) f[i]=f[i]+1;
                gf(f[i]);
                i++;
            }
        }
        else
        {
            scanf("%lld%d",&l,&r);
            if (l>r) swap(l,r);
            printf("%lld\n",gs(r)-gs(l-1));
        }
//      for (int i=1;i<=n;++i) printf("f[%d]= %d\n",i,f[i]);
    }
}