BZOJ3038 上帝造题的七分钟2

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

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

Sample Input

10
1 2 3 4 5 6 7 8 9 10
5
0 1 10
1 1 10
1 1 5
0 5 8
1 4 8

Sample Output

19
7
6

HINT

1：对于100%的数据，1<=n<=100000，1<=l<=r<=n，数列中的数大于0，且不超过1e12。数据不保证L<=R

题解

题目同BZOJ3211 花神游历各国 。用到并查集的思想，跳过1开根。

#include <cstdio>
#include <iostream>
#include <cmath>
#include <stack>
#include <algorithm>
#include <cstring>
#include <climits>
#define MAXN 100000+10
#define LL long long
using namespace std;
LL n,m,a[MAXN],num[MAXN<<2],sit[MAXN];
int fa[MAXN];
int find(int x)
{
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
void build(int rt,int l,int r)
{
    if(l==r)
    {
        num[rt]=a[l];
        sit[l]=rt;
        return ;
    }
    int m=(l+r)>>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    num[rt]=num[rt<<1]+num[rt<<1|1];
}
void update(int x,int l,int r,int rt)
{
    if(l==r&&l==x)
    {
        num[rt]=sqrt(num[rt]);
        return;
    }
    int m=(l+r)>>1;
    if(x<=m) update(x,l,m,rt<<1); else
    if(x>m)  update(x,m+1,r,rt<<1|1);
    num[rt]=num[rt<<1]+num[rt<<1|1];
}
LL query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
        return num[rt];
    int m=(l+r)>>1;LL ans=0;
    if(L<=m) ans+=query(L,R,l,m,rt<<1);
    if(R>m) ans+=query(L,R,m+1,r,rt<<1|1);
    return ans;
}

int main()
{
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        fa[i]=i;
    }
    fa[n+1]=n+1;
    build(1,1,n);
    scanf("%lld",&m);
    for(int i=1;i<=m;i++)
    {
        int o,a,b;
        scanf("%d%d%d",&o,&a,&b);
        if(a>b) swap(a,b);
        if(!o)
        {
            for(int j=find(a-1)+1;j<=find(b);)
            {   
                update(j,1,n,1);
                if(num[sit[j]]==1||num[sit[j]]==0)
                    fa[j-1]=j;
                j=find(j)+1;
            }
        }else
        printf("%lld\n",query(a,b,1,n,1));
    }
    return 0;
}