bzoj 3038 上帝造题的七分钟2

3038: 上帝造题的七分钟2
Time Limit: 3 Sec Memory Limit: 128 MB
Submit: 1124 Solved: 504
[Submit][Status][Discuss]
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。

2：数据不保证L<=R 若L>R，请自行交换L,R，谢谢！

---

【分析】
神题啊神题…
思路：树状数组+并查集维护。
由于数都在int范围内，所以一个数开5、6次根号就是1了。
所以当我们修改的时候，找出该点的右端第一个不是1的点进行暴力修改即可。
所以用并查集维护，如果点i<=1，那么fa[i]=i+1，路径压缩一波就能做了

---

【代码】

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define ll long long
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=100005;
ll n,m,fa[mxn],a[mxn],c[mxn];
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
inline int lowbit(int x) {return x&-x;}
inline int find(int x)
{
    if(x==fa[x] || !fa[x]) return fa[x]=x;
    fa[x]=find(fa[x]);
    return fa[x];
}
inline void update(int x,ll y)
{
    for(int i=x;i<=n;i+=lowbit(i))
      c[i]+=y;
}
inline ll getsum(int x)
{
    ll sum=0;
    for(int i=x;i;i-=lowbit(i))
      sum+=c[i];
    return sum;
}
inline void change(int x,int y)
{
    int i,j;
    for(i=find(x);i<=y;i=find(i+1))
    {
        ll tmp=(ll)sqrt(a[i]);
        update(i,tmp-a[i]);
        a[i]=tmp;
        if(a[i]<=1) fa[i]=find(i+1); 
    }
}
int main()
{
    int i,j;
    ll x,y,p;
    n=read();
    fo(i,1,n)
    {
        scanf("%lld",&a[i]);
        if(a[i]<=1) fa[i]=i+1;
        update(i,a[i]);
    }
    m=read();
    fo(i,1,m)
    {
        p=read(),x=read(),y=read();
        if(x>y) swap(x,y);
        if(p==0) change(x,y);
        else printf("%lld\n",getsum(y)-getsum(x-1));
    }
    return 0;
}