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
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
7
6
HINT
1:对于100%的数据,1<=n<=100000,1<=l<=r<=n,数列中的数大于0,且不超过1e12。
2:数据不保证L<=R 若L>R,请自行交换L,R,谢谢!
Source
并查集维护这个点之后第一个不是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;
}
}
else
printf("%lld\n",query(y)-query(x-1));
}
return 0;
}