一维的区间修改+区间查询,显然可以用线段树来做。
裸的线段树会超时,因为每次开方都要递归到底……(开始将开方的和写成了和的开方QAQ)
剪枝就是对于一个数如果是1或0,那么之后无需再对其进行操作,因为开方不会影响这个数值。
#include<bits/stdc++.h>
#define N 400005
#define ll long long
using namespace std;
int n,m;
int l[N],r[N];
ll sum[N],a[N>>2];
bool flag[N];
inline ll read()
{
ll a=0,f=1; char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
return a*f;
}
inline void pushup(int k)
{
sum[k]=sum[k<<1]+sum[k<<1|1];
flag[k]=flag[k<<1]&flag[k<<1|1];
}
void build(int k,int x,int y)
{
l[k]=x; r[k]=y;
if (l[k]==r[k])
{
sum[k]=a[l[k]];
if (sum[k]==0||sum[k]==1) flag[k]=1;
return;
}
int mid=l[k]+r[k]>>1;
build(k<<1,x,mid); build(k<<1|1,mid+1,y);
pushup(k);
}
void change(int k,int x,int y)
{
if (flag[k]) return;
if (l[k]==r[k])
{
sum[k]=(ll)sqrt(sum[k]);
if (sum[k]==0||sum[k]==1) flag[k]=1;
return;
}
int mid=l[k]+r[k]>>1;
if (y<=mid) change(k<<1,x,y);
else if (x>mid) change(k<<1|1,x,y);
else change(k<<1,x,mid),change(k<<1|1,mid+1,y);
pushup(k);
}
ll query(int k,int x,int y)
{
if (x<=l[k]&&r[k]<=y) return sum[k];
int mid=l[k]+r[k]>>1;
if (y<=mid) return query(k<<1,x,y);
else if (x>mid) return query(k<<1|1,x,y);
else return query(k<<1,x,mid)+query(k<<1|1,mid+1,y);
}
int main()
{
n=read();
for (int i=1;i<=n;i++) a[i]=read();
build(1,1,n);
m=read();
for (int i=1;i<=m;i++)
{
int opt=read(),l=read(),r=read();
if (l>r) swap(l,r);
if (opt==0) change(1,l,r);
else printf("%lld\n",query(1,l,r));
}
return 0;
}