BZOJ 3038: 上帝造题的七分钟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。

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

Source

Poetize4

并查集维护这个点之后第一个不是1的数，树状数组维护sum。

#include<algorithm>#include<iostream>#include<cstdio>#include<cmath>using namespace std;const int N=100005;int n,q,f[N];long long c[N],data[N];int fnd(int x){if(x!=f[x])f[x]=fnd(f[x]);return f[x];}int lowbit(int x){return x&-x;}void add(int pos,long long x){while(pos<=n){c[pos]+=x;pos+=lowbit(pos);}}long long query(int pos){long long sum=0;while(pos>0){sum+=c[pos];pos-=lowbit(pos);}return sum;}int main(){scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%lld",&data[i]);for(int i=1;i<=n;i++){f[i]=i;add(i,data[i]);}f[n+1]=n+1;scanf("%d",&q);while(q--){int k,x,y;scanf("%d%d%d",&k,&x,&y);if(x>y)swap(x,y);if(k==0){for(int i=fnd(x);i<=y;i=fnd(i+1)){long long t=data[i];data[i]=sqrt(data[i]);add(i,data[i]-t);if(data[i]==1)f[i]=i+1;}}elseprintf("%lld\n",query(y)-query(x-1));}return 0;}