bzoj3038 上帝造题的七分钟2

Description

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

第一行一个整数n，代表数列中数的个数。 第二行n个正整数，表示初始状态下数列中的数。 第三行一个整数m，表示有m次操作。
接下来m行每行三个整数k,l,r，k=0表示给[l,r]中的每个数开平方(下取整)，k=1表示询问[l,r]中各个数的和。 Output

对于询问操作，每行输出一个回答。

考虑到实际上需要修改的次数很少，可以用并查集维护从自己开始第一个不是1的数的位置，每次只对不是1的位置进行修改，并且用树状数组维护。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define LL long long
int fa[100010],n,m;
LL a[100010],s[100010];
int find(int x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
void add(int p,LL x)
{
    for (;p<=n;p+=p&-p)
      s[p]+=x;
}
LL qry(int p)
{
    LL ret=0;
    for (;p;p-=p&-p)
      ret+=s[p];
    return ret;
}
int main()
{
    int i,j,k,p,q,l,r,x;
    LL t;
    scanf("%d",&n);
    for (i=1;i<=n;i++)
      scanf("%lld",&a[i]),fa[i]=i,add(i,a[i]);
    fa[n+1]=n+1;
    scanf("%d",&m);
    while (m--)
    {
        scanf("%d%d%d",&x,&l,&r);
        if (l>r) swap(l,r);
        if (x==0)
        {
            for (l=find(l);l<=r;l=find(l+1))
            {
                t=a[l];
                a[l]=sqrt(a[l]);
                add(l,a[l]-t);
                if (a[l]==1) fa[l]=l+1;
            }
        }
        else printf("%lld\n",qry(r)-qry(l-1));
    }
}