BZOJ[3038]上帝造题的7分钟2 树状数组+并查集

题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3038

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，谢谢！

同BZOJ[3211]花神游历各国
处理一下L>R的情况即可…

题解:http://blog.youkuaiyun.com/waduan2/article/details/77434892

代码如下：

#include<algorithm>
#include<ctype.h>
#include<cstdio>
#include<cmath>
#define N 100500
#define int long long
using namespace std;
inline int read(){
    int f=1,x=0;
    char c=getchar();
    while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
    while(isdigit(c)){x=(x<<3)+(x<<1)+c-'0';c=getchar();}
    return x*f;
}
int n,m,k,l,r,t,q[N],a[N],f[N];
int lowbit(int x){return x&-x;}
void add(int x,int v){
    while(x<=n){
        q[x]=q[x]+v;
        x=x+lowbit(x);
    }
}
long long sum(int x){
    long long tmp=0;
    while(x){
        tmp=tmp+q[x];
        x=x-lowbit(x);
    }
    return tmp;
}
int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
main(){
    n=read();
    for(int i=1;i<=n;i++) a[i]=read(),add(i,a[i]),f[i]=a[i]<=1?i+1:i;
    f[n+1]=n+1;
    m=read();
    for(int i=1;i<=m;i++){
        k=read();l=read();r=read();
        if(l>r) swap(l,r);
        if(k==1) printf("%lld\n",sum(r)-sum(l-1));
        else{
            for(int j=l;j<=r;j=find(j+1)){
                int tmp=(int)sqrt(a[j]);
                add(j,tmp-a[j]);
                a[j]=tmp;
                if(a[j]<=1) f[j]=find(j+1);
            }
        }
    }
return 0;
}