BZOJ 3038 上帝造题的七分钟2

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

---

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

---

【代码】

#include <cmath>#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#define ll long longusing 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]);    }}